mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity

Exclude naflex models from jit tests

Browse Source
This commit is contained in:
Ross Wightman 2025-04-08 07:59:19 -07:00
parent 825edccf19
commit 9b23d6dea2
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/test_models.py
View File

@ -78,7 +78,7 @@ else:
EXCLUDE_FILTERS = ['*enormous*'] EXCLUDE_FILTERS = ['*enormous*']
NON_STD_EXCLUDE_FILTERS = ['*gigantic*', '*enormous*', '*_3b_*'] NON_STD_EXCLUDE_FILTERS = ['*gigantic*', '*enormous*', '*_3b_*']
EXCLUDE_JIT_FILTERS = ['hiera_*'] EXCLUDE_JIT_FILTERS = ['hiera_*', '*naflex*']
TARGET_FWD_SIZE = MAX_FWD_SIZE = 384 TARGET_FWD_SIZE = MAX_FWD_SIZE = 384
TARGET_BWD_SIZE = 128 TARGET_BWD_SIZE = 128