Fix CUDA device mismatch (#371)

Making devices for zero or target tensors consistent with inputs.

Signed-off-by: Yaochen Xie <ethanycx@tamu.edu>

Author: ycremar

generative/losses/adversarial_loss.py

@@ -88,7 +88,7 @@ def get_target_tensor(self, input: torch.FloatTensor, target_is_real: bool) -> t
         Returns:
         """
         filling_label = self.real_label if target_is_real else self.fake_label
-        label_tensor = torch.tensor(1).fill_(filling_label).type(input.type())
+        label_tensor = torch.tensor(1).fill_(filling_label).type(input.type()).to(input[0].device)
         label_tensor.requires_grad_(False)
         return label_tensor.expand_as(input)
 
@@ -101,7 +101,7 @@ def get_zero_tensor(self, input: torch.FloatTensor) -> torch.Tensor:
         Returns:
         """
 
-        zero_label_tensor = torch.tensor(0).type(input[0].type())
+        zero_label_tensor = torch.tensor(0).type(input[0].type()).to(input[0].device)
         zero_label_tensor.requires_grad_(False)
         return zero_label_tensor.expand_as(input)