浏览代码

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

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

2
ml-agents/mlagents/trainers/tests/torch/saver/test_saver.py


# asserts convolutions have same parameters after load
for conv1, conv2 in zip(conv_params, conv_params2):
assert torch.equal(conv1, conv2)
# asserts layers still have different dimensions
# asserts layers still have different dimensions
for mod1, mod2 in zip(policy.actor.parameters(), policy2.actor.parameters()):
if mod1.shape[0] == 12:
assert mod2.shape[0] == 10

正在加载...
取消
保存