code: def reset_training_state(self, dones, batch_info):
          
          for idx, done in enumerate(dones):
              if done > 0.5:
                  self.processes[idx].reset()
docs: A hook for a model to react when during training episode is finished