Remove unpacking

This is required when calling grad.

Author: yngtodd

common/darts/architecture.py

@@ -33,9 +33,7 @@ def comp_unrolled_model(self, data, target, eta, optimizer):
         data : torch.tensor
 
         target : torch.tensor
-
         eta : float
-
         optimizer : torch.optim.optimizer
              optimizer of theta, not optimizer of alpha
 
@@ -44,10 +42,11 @@ def comp_unrolled_model(self, data, target, eta, optimizer):
         model_unrolled
         """
         # forward to get loss
-        _, loss = self.model.loss(data, target, reduce='mean')
+        loss = self.model.loss(data, target)
         # flatten current weights
         theta = F.flatten(self.model.parameters()).detach()
         # theta: torch.Size([1930618])
+        # print('theta:', theta.shape)
         try:
             # fetch momentum data from theta optimizer
             moment = F.flatten(optimizer.state[v]['momentum_buffer'] for v in self.model.parameters())