Commit e1224a2a authored by Sylvain Gugger's avatar Sylvain Gugger
Browse files

Making save_load test slow as it times out

parent 0b567aa4
......@@ -328,6 +328,7 @@ class TFViTMAEModelTest(TFModelTesterMixin, unittest.TestCase):
# overwrite from common since TFViTMAEForPretraining has random masking, we need to fix the noise
# to generate masks during test
@slow()
def test_keras_save_load(self):
# make mask reproducible
np.random.seed(2)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment