Fix lars tests

pull/816/head
Ross Wightman 3 years ago
parent a426511c95
commit 42c1f0cf6c

@ -499,7 +499,7 @@ def test_lars(optimizer):
lambda weight, bias: create_optimizer_v2( lambda weight, bias: create_optimizer_v2(
_build_params_dict(weight, bias, lr=1e-3), _build_params_dict(weight, bias, lr=1e-3),
optimizer, optimizer,
lr=1e-1) lr=1e-3)
) )
_test_basic_cases( _test_basic_cases(
lambda weight, bias: create_optimizer_v2( lambda weight, bias: create_optimizer_v2(

Loading…
Cancel
Save