Make more test models smaller (#25005)
* Make more test models tiny * Make more test models tiny * More models * More models
This commit is contained in:
@@ -55,8 +55,8 @@ class CvtModelTester:
|
||||
batch_size=13,
|
||||
image_size=64,
|
||||
num_channels=3,
|
||||
embed_dim=[16, 48, 96],
|
||||
num_heads=[1, 3, 6],
|
||||
embed_dim=[16, 32, 48],
|
||||
num_heads=[1, 2, 3],
|
||||
depth=[1, 2, 10],
|
||||
patch_sizes=[7, 3, 3],
|
||||
patch_stride=[4, 2, 2],
|
||||
@@ -247,10 +247,6 @@ class CvtModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCase):
|
||||
config_and_inputs = self.model_tester.prepare_config_and_inputs()
|
||||
self.model_tester.create_and_check_for_image_classification(*config_and_inputs)
|
||||
|
||||
@unittest.skip("Will be fixed soon by reducing the size of the model used for common tests.")
|
||||
def test_model_is_small(self):
|
||||
pass
|
||||
|
||||
@slow
|
||||
def test_model_from_pretrained(self):
|
||||
for model_name in CVT_PRETRAINED_MODEL_ARCHIVE_LIST[:1]:
|
||||
|
||||
Reference in New Issue
Block a user