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