浏览代码

Add more verbose messages when using --initialize-from and --resume (#5231)

* Add more verbose messages when using --initialize-from and resume

* Fix typo

* Add periods
/check-for-ModelOverriders
GitHub 3 年前
当前提交
1fbbd97b
共有 2 个文件被更改,包括 3 次插入1 次删除
  1. 2
      ml-agents/mlagents/trainers/model_saver/torch_model_saver.py
  2. 2
      ml-agents/mlagents/trainers/stats.py

2
ml-agents/mlagents/trainers/model_saver/torch_model_saver.py


# This argument is mainly for initialization of the ghost trainer's fixed policy.
reset_steps = not self.load
if self.initialize_path is not None:
logger.info(f"Initializing from {self.initialize_path}.")
logger.info(f"Resuming from {self.model_path}.")
self._load_model(self.model_path, policy, reset_global_steps=reset_steps)
def _load_model(

2
ml-agents/mlagents/trainers/stats.py


for file_name in os.listdir(directory_name):
if file_name.startswith("events.out"):
logger.warning(
f"Deleting TensorBoard data {file_name} that was left over from a"
f"Deleting TensorBoard data {file_name} that was left over from a "
"previous run."
)
full_fname = os.path.join(directory_name, file_name)

正在加载...
取消
保存