mirror of
https://github.com/huggingface/pytorch-image-models.git
synced 2025-06-03 15:01:08 +08:00
Keep basic optim test LR range closer to before w/ updated code
This commit is contained in:
parent
269bc084fa
commit
1a70036691
@ -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
|
||||
|
Loading…
x
Reference in New Issue
Block a user