浏览代码

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

Co-authored-by: Vincent-Pierre BERGES <vincentpierre@unity3d.com>
/fix-resume-imi
GitHub 3 年前
当前提交
14b09ba3
共有 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


# ValueError is raised by the optimizer's load_state_dict if the parameters have
# have changed. Note, the optimizer uses a completely different load_state_dict
# function because it is not an nn.Module.
# RuntimeError is raised by PyTorch if there is a size mismatch between modules
# 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 not changed shape.
except (KeyError, ValueError, RuntimeError) as err:

正在加载...
取消
保存