fix: initialization

Author: cowana-ai

lib/SupervisedMetricPretraining.py

@@ -26,7 +26,7 @@ def __init__(self,trainloader,n, t=0.07):
             Temperature parameter as described in https://arxiv.org/pdf/1805.01978.pdf.
 
         """
-        super(Supervised_Pretraining,self).__init__()
+        super(Supervised_Pretraining_Loss,self).__init__()
         # get train labels
         self.labels = get_train_labels(trainloader)
         # Softmax loss