erlich / optimizer

Commit History

  1. add all optimizer states

    76a0236
  2. add ema weights and optimizer state

    d56703f