浏览代码

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

Co-authored-by: Vincent-Pierre BERGES <vincentpierre@unity3d.com>
/fix-resume-imi
GitHub 4 年前
当前提交
5108ec97
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 2
      ml-agents/mlagents/trainers/model_saver/torch_model_saver.py

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


# function because it is not an nn.Module.
# RuntimeError is raised by PyTorch if there is a size mismatch between modules
# of the same name. This will still partially assign values to those layers that
# have no changed shape.
# have not changed shape.
except (KeyError, ValueError, RuntimeError) as err:
logger.warning(f"Failed to load for module {name}. Initializing")
logger.debug(f"Module loading error : {err}")

正在加载...
取消
保存