浏览代码

fix tests

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

2
ml-agents/mlagents/trainers/settings.py


strength: float = 1.0
samples_per_update: int = 0
num_epoch: int = 3
batch_size: int = 256
batch_size: int = 1024
@attr.s(auto_attribs=True)

2
ml-agents/mlagents/trainers/tests/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 == TrainerSettings().hyperparameters.batch_size
assert bc_module.batch_size == 1024
# Assign strange values and see if it overrides properly
bc_settings = BehavioralCloningSettings(
demo_path=os.path.dirname(os.path.abspath(__file__)) + "/" + "test.demo",

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 == 256
assert bc_module.batch_size == 1024
# Assign strange values and see if it overrides properly
bc_settings = BehavioralCloningSettings(
demo_path=os.path.dirname(os.path.abspath(__file__)) + "/" + "test.demo",

正在加载...
取消
保存