def initialize_training(self, training_info: TrainingInfo, model_state=None, hidden_state=None):
if model_state is None:<EOL><INDENT>self.model.reset_weights()<EOL><DEDENT>else:<EOL><INDENT>self.model.load_state_dict(model_state)<EOL><DEDENT>
Prepare for training