Python ReversiModel.create_session Exemples

Exemple #1

Fichier : game_model.py Projet : 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

Exemple #2

Fichier : self_play.py Projet : 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

Exemple #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