浏览代码

Exclude test_visual_encoder_trains from GPU test (#5367)

/colab-links
GitHub 4 年前
当前提交
b767b66b
共有 1 个文件被更改,包括 1 次插入0 次删除
  1. 1
      ml-agents/mlagents/trainers/tests/torch/test_encoders.py

1
ml-agents/mlagents/trainers/tests/torch/test_encoders.py


(FullyConnectedVisualEncoder, 36),
],
)
@pytest.mark.check_environment_trains
def test_visual_encoder_trains(vis_class, size):
torch.manual_seed(0)
image_size = (size, size, 1)

正在加载...
取消
保存