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 703f127 commit 71b8762Copy full SHA for 71b8762
tests/test_gradients.py
@@ -1,7 +1,7 @@
1
import pytest
2
import torch
3
4
-from pytorch_optimizer import SAM, WSAM, AdamP, Lookahead, TRAC, load_optimizer
+from pytorch_optimizer import SAM, TRAC, WSAM, AdamP, Lookahead, load_optimizer
5
from pytorch_optimizer.base.exception import NoSparseGradientError
6
from tests.constants import NO_SPARSE_OPTIMIZERS, SPARSE_OPTIMIZERS, VALID_OPTIMIZER_NAMES
7
from tests.utils import build_environment, simple_parameter, simple_sparse_parameter, sphere_loss
0 commit comments