- An argument updated in
Trainer.fit()
:save_every_ckpt
->save_ckpt_every_k_epochs
- Added
params_sharded
andopt_state_sharded
inTrainer.__init__()
, for memory saving.
Trainer.fit()
: save_every_ckpt
-> save_ckpt_every_k_epochs
params_sharded
and opt_state_sharded
in Trainer.__init__()
, for memory saving.