def __getstate__(self): state = Trainer.__getstate__(self) state["trainer_state"] = self.state.copy() if self.train_exec_impl: state["train_exec_impl"] = ( self.train_exec_impl.shared_metrics.get().save()) return state
def __getstate__(self): state = Trainer.__getstate__(self) state.update({ "num_target_updates": self.num_target_updates, "last_target_update_ts": self.last_target_update_ts, }) return state
def __getstate__(self): state = Trainer.__getstate__(self) state["trainer_state"] = self.state.copy() if self.train_pipeline: state["train_pipeline"] = self.train_pipeline.metrics.save() return state
def __getstate__(self): state = Trainer.__getstate__(self) state["trainer_state"] = self.state.copy() return state
def __getstate__(self): state = Trainer.__getstate__(self) state.update(self.state) return state