예제 #1
0
 def __setstate__(self, state):
     Trainer.__setstate__(self, state)
     self.train_exec_impl.shared_metrics.get().restore(
         state["train_exec_impl"])
예제 #2
0
 def __setstate__(self, state):
     Trainer.__setstate__(self, state)
     self.state = state["trainer_state"].copy()
     if self.train_pipeline:
         self.train_pipeline.metrics.restore(state["train_pipeline"])
예제 #3
0
 def __setstate__(self, state):
     Trainer.__setstate__(self, state)
     self.state = state["trainer_state"].copy()
예제 #4
0
 def __setstate__(self, state):
     Trainer.__setstate__(self, state)
     self.state = state["trainer_state"].copy()
     if self.train_exec_impl:
         self.train_exec_impl.shared_metrics.get().restore(
             state["train_exec_impl"])
예제 #5
0
 def __setstate__(self, state):
     Trainer.__setstate__(self, state)
     self.state = state
예제 #6
0
파일: dqn.py 프로젝트: songhappy/ray 
 def __setstate__(self, state):
     Trainer.__setstate__(self, state)
     self.num_target_updates = state["num_target_updates"]
     self.last_target_update_ts = state["last_target_update_ts"]