Esempi in Python per ReversiModel.create_session

Esempio n. 1

File: game_model.py Progetto: apollo-time/reversi-alpha-zero

    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

Esempio n. 2

File: self_play.py Progetto: apollo-time/reversi-alpha-zero

 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

Esempio n. 3

 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