From ddf3a38c391185e286036f01d513b2927dc2c480 Mon Sep 17 00:00:00 2001
From: Sandhan Aniket Ramdas <138451780+Aniketiitk21@users.noreply.github.com>
Date: Sun, 17 Dec 2023 11:57:49 +0530
Subject: [PATCH] Update main.py

---
 src/main.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/main.py b/src/main.py
index d6b886f..ca05c74 100644
--- a/src/main.py
+++ b/src/main.py
@@ -11,7 +11,7 @@
 from pytorch_lightning.callbacks import ModelCheckpoint, LearningRateMonitor
 
 from utils import SSLDataset, ContrastiveLoss, CosineSimilarity, BiLinearSimilarity, MyCallBack
-from models import CustomArch7, FpNetwork, ProjectionHead1
+from models import CustomArch7, ProjectionHead1
 from train import ContrastiveModel
 
 # os.environ["CUDA_VISIBLE_DEVICES"]="1"