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

Making save_load test slow as it times out

Browse Source
This commit is contained in:
Sylvain Gugger
2022-09-14 10:01:22 -04:00
parent 0b567aa430
commit e1224a2a0f
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
tests/models/vit_mae/test_modeling_tf_vit_mae.py
View File

@@ -328,6 +328,7 @@ class TFViTMAEModelTest(TFModelTesterMixin, unittest.TestCase):
# overwrite from common since TFViTMAEForPretraining has random masking, we need to fix the noise # overwrite from common since TFViTMAEForPretraining has random masking, we need to fix the noise
# to generate masks during test # to generate masks during test
@slow()
def test_keras_save_load(self): def test_keras_save_load(self):
# make mask reproducible # make mask reproducible
np.random.seed(2) np.random.seed(2)