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

CI: fix efficientnet pipeline timeout and prevent future similar issues due to large image size (#33123)

Browse Source 
* fix param not being passed in tested; add exceptions

* better source of model name

* Update utils/create_dummy_models.py

Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>

---------

Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>
This commit is contained in:
Joao Gante
2024-08-27 11:58:27 +01:00
committed by GitHub
parent 3806faa171
commit ab0ac3b98f
4 changed files with 38 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
tests/models/efficientnet/test_modeling_efficientnet.py
View File

@@ -83,6 +83,7 @@ class EfficientNetModelTester:
def get_config(self):
return EfficientNetConfig(
image_size=self.image_size,
num_channels=self.num_channels,
kernel_sizes=self.kernel_sizes,
in_channels=self.in_channels,