update: test_sam_optimizers

kozistr · kozistr · commit 69ca1a09142d · 2022-02-20T18:02:39.000+09:00
diff --git a/tests/test_optimizers.py b/tests/test_optimizers.py
@@ -181,6 +181,7 @@ def test_sam_optimizers(adaptive, optimizer_sam_config):
 
     optimizer_class, config, iterations = optimizer_sam_config
     optimizer = SAM(model.parameters(), optimizer_class, **config, adaptive=adaptive)
+    optimizer.reset()  # meaningless, but for coverage
 
     init_loss, loss = np.inf, np.inf
     for _ in range(iterations):
