浏览代码

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

/develop/torch-save-rp
GitHub 4 年前
当前提交
80b7a6d3
共有 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


saver2 = TorchSaver(trainer_settings, path1, load=True)
saver2.register(policy2)
saver2.register(optimizer2)
saver2.initialize_or_load(policy2)
saver2.initialize_or_load() # This is to load the optimizers
# assert the models have the same weights

正在加载...
取消
保存