def test(self):
self.training = False
if not self.recover:
logger.warning("No pre-trained model has been loaded.")
if self.display_env:
self.monitor.video_callable = MonitorV2.always_call_video
try:
self.agent.eval()
except AttributeError:
pass
self.run_episodes()
self.close()
def load_agent_model(self, model_path):
if model_path is True:
model_path = self.directory / self.SAVED_MODELS_FOLDER / "latest.tar"
try:
self.agent.load(filename=model_path)
logger.info("Load {} model from {}".format(
self.agent.__class__.__name__, model_path))
except FileNotFoundError:
logger.warning(
"No pre-trained model found at the desired location.")
except NotImplementedError:
pass
