Trainer.save_states

Trainer.save_states(self, fname)

Saves trainer states (e.g. optimizer, momentum) to a file.

Parameters:

fname : str

Path to output states file.