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