From 54e90e82a5a6367d468e4f6dd5982715e4e20a72 Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Fri, 27 Aug 2021 10:39:31 -0700 Subject: [PATCH] Another attempt at sgd momentum test passing... --- tests/test_optim.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_optim.py b/tests/test_optim.py index a0fe994e..a46a59f0 100644 --- 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)