Skip to content

Rename TrainState.epoch as TrainState.iteration to better represent i… #1496

Rename TrainState.epoch as TrainState.iteration to better represent i…

Rename TrainState.epoch as TrainState.iteration to better represent i… #1496