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