File tree 2 files changed +4
-11
lines changed
2 files changed +4
-11
lines changed Original file line number Diff line number Diff line change 44
44
45
45
- name : Test examples
46
46
run : |
47
- python examples/openvino/test_examples.py
48
- #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
49
- #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
50
- #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
51
- #python examples/openvino/image-classification/run_image_classification.py --model_name_or_path nateraw/vit-base-beans --dataset_name beans --max_train_samples 10 --max_eval_samples 2 --remove_unused_columns False --do_train --do_eval --learning_rate 2e-5 --num_train_epochs 1 --logging_strategy steps --logging_steps 1 --evaluation_strategy epoch --save_strategy epoch --save_total_limit 1 --seed 1337 --output_dir /tmp/beans_outputs/
52
-
47
+ python examples/openvino/test_examples.py
Original file line number Diff line number Diff line change 17
17
import unittest
18
18
from unittest .mock import patch
19
19
20
- #os.environ["CUDA_VISIBLE_DEVICES"] = ""
21
-
22
20
SRC_DIRS = [
23
21
os .path .join (os .path .dirname (__file__ ), dirname )
24
22
for dirname in [
@@ -42,7 +40,7 @@ def test_audio_classification(self):
42
40
with tempfile .TemporaryDirectory () as tmp_dir :
43
41
test_args = f"""
44
42
run_audio_classification.py
45
- --model_name_or_path facebook/wav2vec2-base
43
+ --model_name_or_path hf-internal-testing/tiny-random-Wav2Vec2Model
46
44
--nncf_compression_config examples/openvino/audio-classification/configs/wav2vec2-base-qat.json
47
45
--dataset_name superb
48
46
--dataset_config_name ks
@@ -99,7 +97,7 @@ def test_text_classification(self):
99
97
with tempfile .TemporaryDirectory () as tmp_dir :
100
98
test_args = f"""
101
99
run_glue.py
102
- --model_name_or_path sshleifer /tiny-distilbert-base-cased-distilled-squad
100
+ --model_name_or_path hf-internal-testing /tiny-random-DistilBertForSequenceClassification
103
101
--task_name sst2
104
102
--max_train_samples 10
105
103
--max_eval_samples 2
@@ -125,7 +123,7 @@ def test_question_answering(self):
125
123
with tempfile .TemporaryDirectory () as tmp_dir :
126
124
test_args = f"""
127
125
run_qa.py
128
- --model_name_or_path sshleifer /tiny-distilbert-base-cased-distilled-squad
126
+ --model_name_or_path hf-internal-testing /tiny-random-DistilBertForQuestionAnswering
129
127
--dataset_name squad
130
128
--do_train
131
129
--do_eval
You can’t perform that action at this time.
0 commit comments