def load_model(self): from reversi_zero.agent.model import ReversiModel model = ReversiModel(self.config) if self.config.opts.new or not load_best_model_weight(model): model.build() save_as_best_model(model) return model
def load_model(self): from reversi_zero.agent.model import ReversiModel model = ReversiModel(self.config) loaded = False if not self.config.opts.new: if self.config.play.use_newest_next_generation_model: loaded = reload_newest_next_generation_model_if_changed(model) or load_best_model_weight(model) else: loaded = load_best_model_weight(model) or reload_newest_next_generation_model_if_changed(model) if not loaded: model.build() save_as_best_model(model) return model