@@ -40,29 +40,29 @@ class TestExamples(unittest.TestCase):
40
40
def test_audio_classification (self ):
41
41
with tempfile .TemporaryDirectory () as tmp_dir :
42
42
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
64
64
--seed 42
65
- --output_dir { tmp_dir }
65
+ --output_dir { tmp_dir }
66
66
--overwrite_output_dir
67
67
""" .split ()
68
68
@@ -72,21 +72,21 @@ def test_audio_classification(self):
72
72
def test_image_classification (self ):
73
73
with tempfile .TemporaryDirectory () as tmp_dir :
74
74
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
81
81
--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
90
90
--seed 1337
91
91
--output_dir { tmp_dir }
92
92
""" .split ()
@@ -96,23 +96,23 @@ def test_image_classification(self):
96
96
97
97
def test_text_classification (self ):
98
98
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
106
106
--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
116
116
--seed 42
117
117
--output_dir { tmp_dir }
118
118
""" .split ()
@@ -123,17 +123,17 @@ def test_text_classification(self):
123
123
def test_question_answering (self ):
124
124
with tempfile .TemporaryDirectory () as tmp_dir :
125
125
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
137
137
--overwrite_output_dir
138
138
--output_dir { tmp_dir }
139
139
""" .split ()
0 commit comments