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

Lower max backward size for tests.

Browse Source
This commit is contained in:
Ross Wightman 2021-05-24 21:36:56 -07:00
parent bfc72f75d3
commit 83487e2a0d
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

@ -32,7 +32,7 @@ else:
TARGET_FWD_SIZE = MAX_FWD_SIZE = 384 TARGET_FWD_SIZE = MAX_FWD_SIZE = 384
TARGET_BWD_SIZE = 128 TARGET_BWD_SIZE = 128
MAX_BWD_SIZE = 384 MAX_BWD_SIZE = 320
MAX_FWD_FEAT_SIZE = 448 MAX_FWD_FEAT_SIZE = 448