error occurred while saving states

[yanghaojin]

thanks a lot for your excellent implementation! I just got an issue when trying to call the save_states() function of gluon trainer after the training finished.

error log: ... return pickle.dumps((self.states, self.optimizer) if dump_optimizer else self.states) _pickle.PicklingError: args[0] from newobj args has the wrong class

[wkcn]

I'm sorry that I reply late.

I have fixed it in https://github.com/wkcn/LookaheadOptimizer-mx/commit/d36ac1d9b4c37e28e7c48120c0c67c8a2b220ddd

Thank you for reporting it : )