浏览代码

fix default bc test

/develop/add-fire/bc
Andrew Cohen 4 年前
当前提交
9f25f53b
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 2
      ml-agents/mlagents/trainers/tests/torch/test_bcmodule.py

2
ml-agents/mlagents/trainers/tests/torch/test_bcmodule.py


)
bc_module = create_bc_module(mock_specs, bc_settings, False, False)
assert bc_module.num_epoch == 3
assert bc_module.batch_size == 1024
assert bc_module.batch_size == TrainerSettings().hyperparameters.batch_size
# Assign strange values and see if it overrides properly
bc_settings = BehavioralCloningSettings(
demo_path=os.path.dirname(os.path.abspath(__file__)) + "/" + "test.demo",

正在加载...
取消
保存