Hi
I would like to same more information on trainer state to load them later, I am doing it like
self.state.A = A
then, I need to load them later as:
path = # saved_path of state
state = TrainerState.load_from_json(path)
print(state.A)
then this gives the error that A does not exist, could you please help me?
thanks @sgugger