Fix a path problem

2 files changed, 1 insertions, 2 deletions

diff --git a/libs/__init__.py b/libs/__init__.py
new file mode 100644
index 0000000..e69de29
--- /dev/null
+++ b/libs/__init__.py
diff --git a/simclr/main.py b/simclr/main.py
index 0252e3a..0f93716 100644
--- a/simclr/main.py
+++ b/simclr/main.py
@@ -11,11 +11,10 @@ from torch.utils.data import Dataset
 from torchvision.datasets import CIFAR10, CIFAR100, ImageNet
 from torchvision.transforms import transforms
 
-from libs.criteria import InfoNCELoss
-
 path = str(Path(Path(__file__).parent.absolute()).parent.absolute())
 sys.path.insert(0, path)
 
+from libs.criteria import InfoNCELoss
 from libs.datautils import color_distortion, Clip, RandomGaussianBlur, TwinTransform
 from libs.optimizers import LARS
 from libs.schedulers import LinearWarmupAndCosineAnneal, LinearLR