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