浏览代码

Update ml-agents/mlagents/trainers/tests/torch/saver/test_saver_reward_providers.py

Co-authored-by: Ruo-Ping (Rachel) Dong <ruoping.dong@unity3d.com>
/develop/torch-save-rp
GitHub 4 年前
当前提交
83e21972
共有 1 个文件被更改,包括 0 次插入1 次删除
  1. 1
      ml-agents/mlagents/trainers/tests/torch/saver/test_saver_reward_providers.py

1
ml-agents/mlagents/trainers/tests/torch/saver/test_saver_reward_providers.py


module2 = module_dict_2[name]
if hasattr(module1, "parameters"):
for param1, param2 in zip(module1.parameters(), module2.parameters()):
print(param2.data)
assert param1.data.ne(param2.data).sum() == 0
正在加载...
取消
保存