def post_optimization_step(self, batch_info, device, model, rollout):
if batch_info.aggregate_batch_number % self.target_update_frequency == <NUM_LIT:0>:<EOL><INDENT>self.target_model.load_state_dict(model.state_dict())<EOL>self.target_model.eval()<EOL><DEDENT>
Steps to take after optimization has been done