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

Keep basic optim test LR range closer to before w/ updated code

Browse Source
This commit is contained in:
Ross Wightman 2024-11-26 13:40:20 -08:00 committed by Ross Wightman
parent 7a165fcb62
commit aeb1ed7a15
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tests/test_optim.py
View File

@ -297,9 +297,9 @@ def test_optim_factory(optimizer):
opt_info = get_optimizer_info(optimizer)
assert isinstance(opt_info, OptimInfo)
lr = (1e-3, 1e-2, 1e-2, 1e-2)
lr = (1e-2,) * 4
if optimizer in ('mars',):
lr = (1e-3, 1e-3, 1e-3, 1e-3)
lr = (1e-3,) * 4
try:
if not opt_info.second_order: # basic tests don't support second order right now