Example #1
0
    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))
Example #2
0
 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)