test.sh is nice now!

czbiohub-sf · Mar 12, 2024 · d8c2cc6 · d8c2cc6
1 parent 285413f
commit d8c2cc6
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/lfm_data_utilities/model_evaluations/yogo/yogo_prediction_handling/test.sh b/lfm_data_utilities/model_evaluations/yogo/yogo_prediction_handling/test.sh
@@ -23,9 +23,9 @@ fi
 model_path=~/celldiagnosis/yogo/trained_models/absurd-moon-607/best.pth
 data_path=~/celldiagnosis/dataset_defs/human-labels/all-dataset-subsets-no-aug.yml
 
-srun --partition=gpu --gres=gpu:1 yogo test $model_path $data_path --wandb --include-mAP --tags handling-test default --PREDICTION-FORMATTER default &
-srun --partition=gpu --gres=gpu:1 yogo test $model_path $data_path --wandb --include-mAP --tags handling-test opt_A --PREDICTION-FORMATTER opt_A &
-srun --partition=gpu --gres=gpu:1 yogo test $model_path $data_path --wandb --include-mAP --tags handling-test opt_B --PREDICTION-FORMATTER opt_B &
+srun --partition=gpu --gres=gpu:1 -c 16 yogo test $model_path $data_path --wandb --include-mAP --tags handling-test default --PREDICTION-FORMATTER default &
+srun --partition=gpu --gres=gpu:1 -c 16 yogo test $model_path $data_path --wandb --include-mAP --tags handling-test opt_A --PREDICTION-FORMATTER opt_A &
+srun --partition=gpu --gres=gpu:1 -c 16 yogo test $model_path $data_path --wandb --include-mAP --tags handling-test opt_B --PREDICTION-FORMATTER opt_B &
 
 wait