diff --git a/.github/workflows/batch-experiment.yaml b/.github/workflows/batch-experiment.yaml index 4af78dbf2..192c6eef9 100644 --- a/.github/workflows/batch-experiment.yaml +++ b/.github/workflows/batch-experiment.yaml @@ -40,29 +40,29 @@ jobs: # Array of tests to exclude exclude_tests=( - "models/mnist/test_mnist.py" - "models/MobileNetV2/test_MobileNetV2.py" - "models/openpose/test_openpose_v2.py" - "models/resnet/test_resnet.py" - "models/resnet50/test_resnet50.py" - "models/roberta/test_roberta.py" - #"models/hand_landmark/test_hand_landmark.py" - "models/squeeze_bert/test_squeeze_bert.py" + #"models/mnist/test_mnist.py" + #"models/MobileNetV2/test_MobileNetV2.py" + #"models/openpose/test_openpose_v2.py" + #"models/resnet/test_resnet.py" + #"models/resnet50/test_resnet50.py" + #"models/roberta/test_roberta.py" + ##"models/hand_landmark/test_hand_landmark.py" + #"models/squeeze_bert/test_squeeze_bert.py" #"models/llama/test_llama.py" #"models/timm/test_timm_image_classification.py" - "models/torchvision/test_torchvision_image_classification.py" + #"models/torchvision/test_torchvision_image_classification.py" #"models/unet/test_unet.py" - "models/albert/test_albert_question_answering.py" - "models/albert/test_alert_sequence_classification.py" - "models/albert/test_albert_token_classification.py" + #"models/albert/test_albert_question_answering.py" + #"models/albert/test_alert_sequence_classification.py" + #"models/albert/test_albert_token_classification.py" #"models/albert/test_albert_masked_lm.py" - "models/unet_carvana/test_unet_carvana.py" - "models/unet_brain/test_unet_brain.py" - "models/autoencoder_linear/test_autoencoder_linear.py" + #"models/unet_carvana/test_unet_carvana.py" + #"models/unet_brain/test_unet_brain.py" + #"models/autoencoder_linear/test_autoencoder_linear.py" #"models/perceiver_io/test_perceiver_io.py" - "models/dpr/test_dpr.py" - "models/bloom/test_bloom.py" - "models/distilbert/test_distilbert.py" + #"models/dpr/test_dpr.py" + #"models/bloom/test_bloom.py" + #"models/distilbert/test_distilbert.py" ) @@ -123,7 +123,7 @@ jobs: fi local batch_size_to_test=$(( (batch_range[0] + batch_range[1]) / 2 )) if (( batch_size_to_test % 2 != 0)); then - batch_size_to_test=(( $batch_size_to_test-1 )) + (( batch_size_to_test -= 1 )) fi echo "Testing with batch size $batch_size_to_test"