浏览代码

add debug with error print out

/fix-resume-imi
Andrew Cohen 3 年前
当前提交
1fb2fa73
共有 1 个文件被更改,包括 4 次插入3 次删除
  1. 7
      ml-agents/mlagents/trainers/model_saver/torch_model_saver.py

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


)
if missing_keys:
logger.warning(
f"Did not find these keys {missing_keys} in checkpoint. Initializing"
f"Did not find these keys {missing_keys} in checkpoint. Initializing."
f"Did not expect these keys {unexpected_keys} in checkpoint. Ignoring"
f"Did not expect these keys {unexpected_keys} in checkpoint. Ignoring."
except (KeyError, TypeError):
except (KeyError, TypeError, RuntimeError) as err:
logger.debug(f"Module loading error : {err}")
if reset_global_steps:
policy.set_step(0)

正在加载...
取消
保存