update: test_sam_methods

Author: kozistr

tests/test_optimizer_parameters.py

@@ -148,6 +148,15 @@ def test_lookahead_parameters():
         Lookahead(optimizer, pullback_momentum='invalid')
 
 
+def test_sam_methods():
+    model: nn.Module = Example()
+    parameters = model.parameters()
+
+    optimizer = SAM(parameters, load_optimizers('adamp'))
+    optimizer.reset()
+    optimizer.load_state_dict(optimizer.state_dict())
+
+
 def test_safe_fp16_methods():
     model: nn.Module = Example()
 

tests/test_optimizers.py

@@ -181,7 +181,6 @@ 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):