From 675d707b9314eb1e435fae9825e4d45ea249c582 Mon Sep 17 00:00:00 2001 From: Christoph Gerum Date: Tue, 28 Nov 2023 17:05:05 +0100 Subject: [PATCH] CI: fix ofa related tests --- test/test_train.py | 19 ------------------- 1 file changed, 19 deletions(-) diff --git a/test/test_train.py b/test/test_train.py index d02b6994..c713a3d0 100644 --- a/test/test_train.py +++ b/test/test_train.py @@ -135,24 +135,6 @@ def test_kitti(): subprocess.run(command_line, shell=True, check=True, cwd=topdir) -@pytest.mark.integration -@pytest.mark.parametrize( - "model,epochs,random_evaluate,random_evaluate_number", - [ - ("ofa_quant", "1", "True", "5"), - ("ofa", "1", "True", "5"), - ("ofa_quant", "1", "False", "5"), - ("ofa", "1", "False", "5"), - ], -) -def test_ofa(model, epochs, random_evaluate, random_evaluate_number): - epochs = 1 - command_line = f"python -m hannah.tools.train --config-name nas_ofa trainer.limit_train_batches=5 trainer.limit_val_batches=5 trainer.limit_test_batches=5 experiment_id=test_ofa nas.model_trainer.epochs_warmup={epochs} nas.model_trainer.epochs_kernel_step={epochs} nas.model_trainer.epochs_depth_step={epochs} nas.model_trainer.epochs_dilation_step={epochs} nas.model_trainer.epochs_width_step={epochs} nas.model_trainer.random_evaluate=False model={model} nas.model_trainer.random_evaluate={random_evaluate} nas.model_trainer.random_eval_number={random_evaluate_number}" - - logging.info("runing commandline %s", command_line) - subprocess.run(command_line, shell=True, check=True, cwd=topdir) - - @pytest.mark.integration @pytest.mark.parametrize( "config", @@ -169,7 +151,6 @@ def test_ofa(model, epochs, random_evaluate, random_evaluate_number): ], ) def test_directional(config): - command_line = f"hannah-train --config-name {config} trainer.fast_dev_run=true module.batch_size=2" subprocess.run(command_line, shell=True, check=True, cwd=topdir)