From 45dafa2970557757857025bf6053f0d29f9cbc85 Mon Sep 17 00:00:00 2001 From: Nooshin Nabizadeh Date: Mon, 18 Mar 2024 10:55:21 -0700 Subject: [PATCH] added test for ov examples v7 --- .github/workflows/test_openvino_examples.yml | 2 +- examples/openvino/{test_ov_examples.py => test_examples.py} | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) rename examples/openvino/{test_ov_examples.py => test_examples.py} (97%) diff --git a/.github/workflows/test_openvino_examples.yml b/.github/workflows/test_openvino_examples.yml index 1c9a6208e4..054ed818fb 100644 --- a/.github/workflows/test_openvino_examples.yml +++ b/.github/workflows/test_openvino_examples.yml @@ -44,7 +44,7 @@ jobs: - name: Test examples run: | - python examples/openvino/test_ov_examples.py + python examples/openvino/test_examples.py #python examples/openvino/audio-classification/run_audio_classification.py --model_name_or_path facebook/wav2vec2-base --nncf_compression_config examples/openvino/audio-classification/configs/wav2vec2-base-qat.json --dataset_name superb --dataset_config_name ks --max_train_samples 10 --max_eval_samples 2 --output_dir /tmp/qat-wav2vec2-base-ft-keyword-spotting --overwrite_output_dir --remove_unused_columns False --do_train --do_eval --learning_rate 3e-5 --max_length_seconds 1 --attention_mask False --warmup_ratio 0.1 --num_train_epochs 1 --gradient_accumulation_steps 1 --dataloader_num_workers 1 --logging_strategy steps --logging_steps 1 --evaluation_strategy epoch --save_strategy epoch --load_best_model_at_end False --seed 42 #python examples/openvino/text-classification/run_glue.py --model_name_or_path sshleifer/tiny-distilbert-base-cased-distilled-squad --task_name sst2 --max_train_samples 10 --max_eval_samples 2 --output_dir /tmp/qat-bert-base-ft-sst2 --overwrite_output_dir --do_train --do_eval --max_seq_length 128 --learning_rate 1e-5 --optim adamw_torch --num_train_epochs 1 --logging_steps 1 --evaluation_strategy steps --eval_steps 1 --save_strategy epoch --seed 42 #python examples/openvino/question-answering/run_qa.py --model_name_or_path sshleifer/tiny-distilbert-base-cased-distilled-squad --dataset_name squad --do_train --do_eval --max_train_samples 10 --max_eval_samples 2 --learning_rate 3e-5 --num_train_epochs 1 --max_seq_length 384 --doc_stride 128 --output_dir /tmp/outputs_squad/ --overwrite_output_dir diff --git a/examples/openvino/test_ov_examples.py b/examples/openvino/test_examples.py similarity index 97% rename from examples/openvino/test_ov_examples.py rename to examples/openvino/test_examples.py index d88876500a..835ee02455 100644 --- a/examples/openvino/test_ov_examples.py +++ b/examples/openvino/test_examples.py @@ -43,7 +43,7 @@ def test_audio_classification(self): test_args = f""" run_audio_classification.py --model_name_or_path facebook/wav2vec2-base - --nncf_compression_config audio-classification/configs/wav2vec2-base-qat.json + --nncf_compression_config examples/openvino/audio-classification/configs/wav2vec2-base-qat.json --dataset_name superb --dataset_config_name ks --max_train_samples 10