Another attempt at sgd momentum test passing...

rwightman · rwightman · commit 54e90e82a5a6 · 2021-09-03T20:50:26.000-07:00
diff --git a/tests/test_optim.py b/tests/test_optim.py
@@ -317,10 +317,10 @@ def test_sgd(optimizer):
     #      lambda opt: ReduceLROnPlateau(opt)]
     # )
     _test_basic_cases(
-        lambda weight, bias: create_optimizer_v2([weight, bias], optimizer, lr=1e-3, momentum=1)
+        lambda weight, bias: create_optimizer_v2([weight, bias], optimizer, lr=3e-3, momentum=1)
     )
     _test_basic_cases(
-        lambda weight, bias: create_optimizer_v2([weight, bias], optimizer, lr=1e-3, momentum=1, weight_decay=.1)
+        lambda weight, bias: create_optimizer_v2([weight, bias], optimizer, lr=3e-3, momentum=1, weight_decay=.1)
     )
     _test_rosenbrock(
         lambda params: create_optimizer_v2(params, optimizer, lr=1e-3)

Original file line number	Diff line number	Diff line change
`@@ -317,10 +317,10 @@ def test_sgd(optimizer):`
`317`	`317`	`# lambda opt: ReduceLROnPlateau(opt)]`
`318`	`318`	`# )`
`319`	`319`	`_test_basic_cases(`
`320`		`- lambda weight, bias: create_optimizer_v2([weight, bias], optimizer, lr=1e-3, momentum=1)`
	`320`	`+ lambda weight, bias: create_optimizer_v2([weight, bias], optimizer, lr=3e-3, momentum=1)`
`321`	`321`	`)`
`322`	`322`	`_test_basic_cases(`
`323`		`- lambda weight, bias: create_optimizer_v2([weight, bias], optimizer, lr=1e-3, momentum=1, weight_decay=.1)`
	`323`	`+ lambda weight, bias: create_optimizer_v2([weight, bias], optimizer, lr=3e-3, momentum=1, weight_decay=.1)`
`324`	`324`	`)`
`325`	`325`	`_test_rosenbrock(`
`326`	`326`	`lambda params: create_optimizer_v2(params, optimizer, lr=1e-3)`