Update main.py

madhavlab · Dec 17, 2023 · ddf3a38 · ddf3a38
1 parent 8ec0f18
commit ddf3a38
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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"