def save(self, name: Optional[str] = None) -> None: if not self._save_dir_path.exists(): Utils.create_directory(self._save_dir_path) if name is None: name = f'{type(self).__name__}_{Utils.get_now_as_str()}' name += '.pt' torch.save(self._policy_net.state_dict(), self._save_dir_path.joinpath(name))
def _create_default_log(self) -> None: log_dir_name = 'log' project_path = Utils.get_base_dir() log_dir_path = project_path.joinpath(log_dir_name) Utils.create_directory(log_dir_path) self._create_unique_log_file(log_dir_path)