Skip to content

Commit f2d8beb

Browse files
echarlaixecharlaix
committed
fix
1 parent 55c481a commit f2d8beb

File tree

1 file changed

+63
-63
lines changed

1 file changed

+63
-63
lines changed

‎examples/openvino/test_examples.py

+63-63
Original file line numberDiff line numberDiff line change
@@ -40,29 +40,29 @@ class TestExamples(unittest.TestCase):
4040
def test_audio_classification(self):
4141
with tempfile.TemporaryDirectory() as tmp_dir:
4242
test_args = f"""
43-
run_audio_classification.py
44-
--model_name_or_path hf-internal-testing/tiny-random-Wav2Vec2Model
45-
--nncf_compression_config examples/openvino/audio-classification/configs/wav2vec2-base-qat.json
46-
--dataset_name superb
47-
--dataset_config_name ks
48-
--max_train_samples 10
49-
--max_eval_samples 2
50-
--remove_unused_columns False
51-
--do_train
52-
--learning_rate 3e-5
53-
--max_length_seconds 1
54-
--attention_mask False
55-
--warmup_ratio 0.1
56-
--num_train_epochs 1
57-
--gradient_accumulation_steps 1
58-
--dataloader_num_workers 1
59-
--logging_strategy steps
60-
--logging_steps 1
61-
--evaluation_strategy epoch
62-
--save_strategy epoch
63-
--load_best_model_at_end False
43+
run_audio_classification.py
44+
--model_name_or_path hf-internal-testing/tiny-random-Wav2Vec2Model
45+
--nncf_compression_config examples/openvino/audio-classification/configs/wav2vec2-base-qat.json
46+
--dataset_name superb
47+
--dataset_config_name ks
48+
--max_train_samples 10
49+
--max_eval_samples 2
50+
--remove_unused_columns False
51+
--do_train
52+
--learning_rate 3e-5
53+
--max_length_seconds 1
54+
--attention_mask False
55+
--warmup_ratio 0.1
56+
--num_train_epochs 1
57+
--gradient_accumulation_steps 1
58+
--dataloader_num_workers 1
59+
--logging_strategy steps
60+
--logging_steps 1
61+
--evaluation_strategy epoch
62+
--save_strategy epoch
63+
--load_best_model_at_end False
6464
--seed 42
65-
--output_dir {tmp_dir}
65+
--output_dir {tmp_dir}
6666
--overwrite_output_dir
6767
""".split()
6868

@@ -72,21 +72,21 @@ def test_audio_classification(self):
7272
def test_image_classification(self):
7373
with tempfile.TemporaryDirectory() as tmp_dir:
7474
test_args = f"""
75-
run_image_classification.py
76-
--model_name_or_path nateraw/vit-base-beans
77-
--dataset_name beans
78-
--max_train_samples 10
79-
--max_eval_samples 2
80-
--remove_unused_columns False
75+
run_image_classification.py
76+
--model_name_or_path nateraw/vit-base-beans
77+
--dataset_name beans
78+
--max_train_samples 10
79+
--max_eval_samples 2
80+
--remove_unused_columns False
8181
--do_train
82-
--do_eval
83-
--learning_rate 2e-5
84-
--num_train_epochs 1
85-
--logging_strategy steps
86-
--logging_steps 1
87-
--evaluation_strategy epoch
88-
--save_strategy epoch
89-
--save_total_limit 1
82+
--do_eval
83+
--learning_rate 2e-5
84+
--num_train_epochs 1
85+
--logging_strategy steps
86+
--logging_steps 1
87+
--evaluation_strategy epoch
88+
--save_strategy epoch
89+
--save_total_limit 1
9090
--seed 1337
9191
--output_dir {tmp_dir}
9292
""".split()
@@ -96,23 +96,23 @@ def test_image_classification(self):
9696

9797
def test_text_classification(self):
9898
with tempfile.TemporaryDirectory() as tmp_dir:
99-
test_args = f"""
100-
run_glue.py
101-
--model_name_or_path hf-internal-testing/tiny-random-DistilBertForSequenceClassification
102-
--task_name sst2
103-
--max_train_samples 10
104-
--max_eval_samples 2
105-
--overwrite_output_dir
99+
test_args = f"""
100+
run_glue.py
101+
--model_name_or_path hf-internal-testing/tiny-random-DistilBertForSequenceClassification
102+
--task_name sst2
103+
--max_train_samples 10
104+
--max_eval_samples 2
105+
--overwrite_output_dir
106106
--do_train
107-
--do_eval
108-
--max_seq_length 128
109-
--learning_rate 1e-5
110-
--optim adamw_torch
111-
--num_train_epochs 1
112-
--logging_steps 1
113-
--evaluation_strategy steps
114-
--eval_steps 1
115-
--save_strategy epoch
107+
--do_eval
108+
--max_seq_length 128
109+
--learning_rate 1e-5
110+
--optim adamw_torch
111+
--num_train_epochs 1
112+
--logging_steps 1
113+
--evaluation_strategy steps
114+
--eval_steps 1
115+
--save_strategy epoch
116116
--seed 42
117117
--output_dir {tmp_dir}
118118
""".split()
@@ -123,17 +123,17 @@ def test_text_classification(self):
123123
def test_question_answering(self):
124124
with tempfile.TemporaryDirectory() as tmp_dir:
125125
test_args = f"""
126-
run_qa.py
127-
--model_name_or_path hf-internal-testing/tiny-random-DistilBertForQuestionAnswering
128-
--dataset_name squad
129-
--do_train
130-
--do_eval
131-
--max_train_samples 10
132-
--max_eval_samples 2
133-
--learning_rate 3e-5
134-
--num_train_epochs 1
135-
--max_seq_length 384
136-
--doc_stride 128
126+
run_qa.py
127+
--model_name_or_path hf-internal-testing/tiny-random-DistilBertForQuestionAnswering
128+
--dataset_name squad
129+
--do_train
130+
--do_eval
131+
--max_train_samples 10
132+
--max_eval_samples 2
133+
--learning_rate 3e-5
134+
--num_train_epochs 1
135+
--max_seq_length 384
136+
--doc_stride 128
137137
--overwrite_output_dir
138138
--output_dir {tmp_dir}
139139
""".split()

0 commit comments

Comments
 (0)