autopep8 fix

pytorch · Nov 27, 2024 · 52f5154 · 52f5154
1 parent 99d807a
commit 52f5154
Show file tree

Hide file tree

Showing 5 changed files with 5 additions and 5 deletions.
diff --git a/examples/cifar10/main.py b/examples/cifar10/main.py
@@ -7,8 +7,8 @@
 import torch.nn as nn
 import torch.optim as optim
 import utils
-from torch.cuda.amp import GradScaler
 from torch.amp import autocast
+from torch.cuda.amp import GradScaler
 
 import ignite
 import ignite.distributed as idist

diff --git a/examples/cifar100_amp_benchmark/benchmark_torch_cuda_amp.py b/examples/cifar100_amp_benchmark/benchmark_torch_cuda_amp.py
@@ -1,7 +1,7 @@
 import fire
 import torch
-from torch.cuda.amp import GradScaler
 from torch.amp import autocast
+from torch.cuda.amp import GradScaler
 from torch.nn import CrossEntropyLoss
 from torch.optim import SGD
 from torchvision.models import wide_resnet50_2

diff --git a/examples/cifar10_qat/main.py b/examples/cifar10_qat/main.py
@@ -6,8 +6,8 @@
 import torch.nn as nn
 import torch.optim as optim
 import utils
-from torch.cuda.amp import GradScaler
 from torch.amp import autocast
+from torch.cuda.amp import GradScaler
 
 import ignite
 import ignite.distributed as idist

diff --git a/examples/references/classification/imagenet/main.py b/examples/references/classification/imagenet/main.py
@@ -6,8 +6,8 @@
 import torch
 
 try:
-    from torch.cuda.amp import GradScaler
     from torch.amp import autocast
+    from torch.cuda.amp import GradScaler
 except ImportError:
     raise RuntimeError("Please, use recent PyTorch version, e.g. >=1.6.0")
 

diff --git a/examples/references/segmentation/pascal_voc2012/main.py b/examples/references/segmentation/pascal_voc2012/main.py
@@ -6,8 +6,8 @@
 import torch
 
 try:
-    from torch.cuda.amp import GradScaler
     from torch.amp import autocast
+    from torch.cuda.amp import GradScaler
 except ImportError:
     raise RuntimeError("Please, use recent PyTorch version, e.g. >=1.6.0")