ndarray
autograd
loss
Trainer.
save_states
Saves trainer states (e.g. optimizer, momentum) to a file.
fname (str) – Path to output states file.
Note
optimizer.param_dict, which contains Parameter information (such as lr_mult and wd_mult) will not be saved.