1
0
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:
Ross Wightman 2024-11-26 13:40:20 -08:00
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