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

Skip wav2vec2 hubert high mem tests (#21643)

Browse Source 
* Skip high memory tests

* Skip high memory tests

* Remove unused import
This commit is contained in:
amyeroberts
2023-02-15 14:17:26 +00:00
committed by GitHub
parent e3d832ff87
commit fc28c006a6
2 changed files with 12 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
tests/models/hubert/test_modeling_tf_hubert.py
View File

@@ -450,19 +450,15 @@ class TFHubertRobustModelTest(TFModelTesterMixin, unittest.TestCase):
model = TFHubertModel.from_pretrained("facebook/hubert-large-ls960-ft") model = TFHubertModel.from_pretrained("facebook/hubert-large-ls960-ft")
self.assertIsNotNone(model) self.assertIsNotNone(model)
# We override here as passing a full batch of 13 samples results in OOM errors for CTC @unittest.skip(reason="Fix me! Hubert hits OOM errors when loss is computed on full batch")
def test_dataset_conversion(self): def test_dataset_conversion(self):
default_batch_size = self.model_tester.batch_size # TODO: (Amy) - check whether skipping CTC model resolves this issue and possible resolutions for CTC
self.model_tester.batch_size = 2 pass
super().test_dataset_conversion()
self.model_tester.batch_size = default_batch_size
# We override here as passing a full batch of 13 samples results in OOM errors for CTC @unittest.skip(reason="Fix me! Hubert hits OOM errors when loss is computed on full batch")
def test_keras_fit(self): def test_keras_fit(self):
default_batch_size = self.model_tester.batch_size # TODO: (Amy) - check whether skipping CTC model resolves this issue and possible resolutions for CTC
self.model_tester.batch_size = 2 pass
super().test_keras_fit()
self.model_tester.batch_size = default_batch_size
@require_tf @require_tf

17
tests/models/wav2vec2/test_modeling_tf_wav2vec2.py
View File

@@ -385,20 +385,15 @@ class TFWav2Vec2ModelTest(TFModelTesterMixin, unittest.TestCase):
model = TFWav2Vec2Model.from_pretrained("facebook/wav2vec2-base-960h") model = TFWav2Vec2Model.from_pretrained("facebook/wav2vec2-base-960h")
self.assertIsNotNone(model) self.assertIsNotNone(model)
# We override here as passing a full batch of 13 samples results in OOM errors for CTC @unittest.skip(reason="Fix me! Wav2Vec2 hits OOM errors when loss is computed on full batch")
@unittest.skip("Fix me!")
def test_dataset_conversion(self): def test_dataset_conversion(self):
default_batch_size = self.model_tester.batch_size # TODO: (Amy) - check whether skipping CTC model resolves this issue and possible resolutions for CTC
self.model_tester.batch_size = 2 pass
super().test_dataset_conversion()
self.model_tester.batch_size = default_batch_size
# We override here as passing a full batch of 13 samples results in OOM errors for CTC @unittest.skip(reason="Fix me! Wav2Vec2 hits OOM errors when loss is computed on full batch")
def test_keras_fit(self): def test_keras_fit(self):
default_batch_size = self.model_tester.batch_size # TODO: (Amy) - check whether skipping CTC model resolves this issue and possible resolutions for CTC
self.model_tester.batch_size = 2 pass
super().test_keras_fit()
self.model_tester.batch_size = default_batch_size
@require_tf @require_tf