fix: used as a cost function

Author: cowana-ai

lib/SupervisedMetricPretraining.py

@@ -13,7 +13,7 @@ def get_train_labels(trainloader, device='cuda'):
         assert all(train_labels != -1)
         train_labels_ = train_labels.to(device)
     return train_labels_
-class Supervised_Pretraining(object):
+class Supervised_Pretraining_Loss(object):
     def __init__(self,trainloader,n, t=0.07):
         """
         Parameters