def load_model(config: Config):
from reversi_zero.agent.model import ReversiModel
model = ReversiModel(config)
if 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:
raise RuntimeError("No models found!")
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
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)
rc = self.config.resource
dirs = get_next_generation_model_dirs(rc)
if not dirs:
logger.debug(f"loading best model")
if not load_best_model_weight(model):
raise RuntimeError(f"Best model can not loaded!")
else:
latest_dir = dirs[-1]
logger.debug(f"loading latest model")
config_path = os.path.join(latest_dir, rc.next_generation_model_config_filename)
weight_path = os.path.join(latest_dir, rc.next_generation_model_weight_filename)
model.load(config_path, weight_path)
return model
def load_best_model(self):
model = ReversiModel(self.config)
load_best_model_weight(model)
return model
def _load_model(self):
from reversi_zero.agent.model import ReversiModel
model = ReversiModel(self.config)
if not load_best_model_weight(model):
raise RuntimeError("best model not found!")
return model
