imports

Lightning-Universe · Nov 19, 2020 · 0d71c09 · 0d71c09
1 parent 7f46efd
commit 0d71c09
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 3 deletions.
diff --git a/pl_bolts/models/self_supervised/simclr/simclr_finetuner.py b/pl_bolts/models/self_supervised/simclr/simclr_finetuner.py
@@ -3,9 +3,9 @@
 
 import pytorch_lightning as pl
 
-from pl_bolts.models.self_supervised.ssl_finetuner import SSLFineTuner
 from pl_bolts.models.self_supervised.simclr.simclr_module import SimCLR
 from pl_bolts.models.self_supervised.simclr.transforms import SimCLRFinetuneTransform
+from pl_bolts.models.self_supervised.ssl_finetuner import SSLFineTuner
 from pl_bolts.transforms.dataset_normalizations import (
     imagenet_normalization,
     stl10_normalization,

diff --git a/pl_bolts/models/self_supervised/simclr/simclr_module.py b/pl_bolts/models/self_supervised/simclr/simclr_module.py
@@ -1,5 +1,4 @@
 import math
-import os
 from argparse import ArgumentParser
 from typing import Callable, Optional
 
@@ -8,7 +7,6 @@
 import torch
 import torch.distributed as dist
 import torch.nn.functional as F
-
 from pytorch_lightning.utilities import AMPType
 from torch import nn
 from torch.optim.optimizer import Optimizer