def _load_model(self, model_dir): from reversi_zero.agent.model import ReversiModel model = ReversiModel(self.config) model.create_session() model.load(model_dir) return model
def load_model(self): from reversi_zero.agent.model import ReversiModel model = ReversiModel(self.config) rc = self.config.resource model.create_session() model.load(rc.model_dir) return model
def load_model(self): from reversi_zero.agent.model import ReversiModel model = ReversiModel(self.config) model.build_train(self.config.resource.tensor_log_dir) model.create_session() logger.debug(f"loading model") steps = model.load(self.config.resource.model_dir) if steps is None: steps = 0 return model, steps