test: byol pytorch should use epochs
`test_byol_pytorch_distributed` uses batches but the model code has an explicit dependency on epochs since we try to access it from `hparams["searcher"]["max_length"]["epochs"]`.
@@ -170,6 +170,6 @@ def test_word_language_lstm_const() -> None:
@pytest.mark.gpu_required
def test_byol_pytorch_distributed() -> None:
config = conf.load_config(conf.cv_examples_path("byol_pytorch/distributed-stl10.yaml"))
- config = conf.set_max_length(config, {"batches": 200})
+ config = conf.set_max_length(config, {"epochs": 1})
exp.run_basic_test_with_temp_config(config, conf.cv_examples_path("byol_pytorch"), 1)
