SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

[Data2Vec] Speed up test (#17660)

Browse Source
This commit is contained in:
Patrick von Platen
2022-06-10 18:48:58 +02:00
committed by GitHub
parent 5e428b71b4
commit b4eef63a1d
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/models/data2vec/test_modeling_data2vec_audio.py
View File

@@ -535,7 +535,7 @@ class Data2VecAudioModelTest(ModelTesterMixin, unittest.TestCase):
def test_mask_feature_prob_ctc(self): def test_mask_feature_prob_ctc(self):
model = Data2VecAudioForCTC.from_pretrained( model = Data2VecAudioForCTC.from_pretrained(
"facebook/data2vec-audio-base-960h", mask_feature_prob=0.2, mask_feature_length=2 "hf-internal-testing/tiny-random-data2vec-seq-class", mask_feature_prob=0.2, mask_feature_length=2
) )
model.to(torch_device).train() model.to(torch_device).train()
processor = Wav2Vec2Processor.from_pretrained( processor = Wav2Vec2Processor.from_pretrained(
@@ -554,7 +554,7 @@ class Data2VecAudioModelTest(ModelTesterMixin, unittest.TestCase):
attention_mask=batch["attention_mask"].to(torch_device), attention_mask=batch["attention_mask"].to(torch_device),
).logits ).logits
self.assertEqual(logits.shape, (4, 299, 32)) self.assertEqual(logits.shape, (4, 1498, 32))
def test_mask_time_prob_ctc(self): def test_mask_time_prob_ctc(self):
model = Data2VecAudioForCTC.from_pretrained( model = Data2VecAudioForCTC.from_pretrained(