浏览代码

use debug statements

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

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


logger.warning(f"Failed to load directly for module {name}.")
for mod_element in mod.state_dict():
try:
logger.warning(f"Copying {mod_element}")
logger.debug(f"Copying {mod_element}")
logger.warning(
logger.debug(
f"{mod_element} was not found or is not loadable (changed shape). Initializing."
)
else:

正在加载...
取消
保存