We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 9c86853 commit 522ca7bCopy full SHA for 522ca7b
tests/test_optimizers.py
@@ -199,7 +199,7 @@ def test_f16_optimizers(optimizer_fp16_config):
199
200
@pytest.mark.parametrize('adaptive', (False, True))
201
@pytest.mark.parametrize('optimizer_sam_config', FP32_OPTIMIZERS, ids=ids)
202
-def test_sam_optimizers(adaptive_type, optimizer_sam_config):
+def test_sam_optimizers(adaptive, optimizer_sam_config):
203
torch.manual_seed(42)
204
205
x_data, y_data = make_dataset()
0 commit comments