meta-SR

seongmin-kye · Sep 16, 2020 · b4c1ea1 · b4c1ea1
1 parent b5434cb
commit b4c1ea1
Show file tree

Hide file tree

Showing 2 changed files with 87 additions and 76 deletions.
diff --git a/.idea/workspace.xml b/.idea/workspace.xml
diff --git a/losses/prototypical.py b/losses/prototypical.py
@@ -6,6 +6,7 @@ class Prototypical(nn.Module):
     def __init__(self):
         super(Prototypical, self).__init__()
 
+        self.zero = torch.tensor(0).cuda()
         self.criterion = torch.nn.CrossEntropyLoss()
         print('Initialized Prototypical Loss')
 
@@ -22,8 +23,8 @@ def forward(self, support, query, label_g, label_e, model, use_GC=True):
         loss_e = self.criterion(logit_e, label_e)
         acc_e = self.accuracy(logit_e, label_e)
 
-        loss_g = 0
-        acc_g = 0
+        loss_g = self.zero
+        acc_g = self.zero
         if use_GC:
             inputs = torch.cat((support, query), dim=0)
             logit_g = F.linear(inputs, F.normalize(model.weight))