edit nn -> F(relu in D_loss)

grey-eye · Jul 4, 2019 · 0bdf308 · 0bdf308
1 parent 40d88e7
commit 0bdf308
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/network/loss.py b/network/loss.py
@@ -66,4 +66,4 @@ def __init__(self, device):
         self.zero = torch.zeros([1]).type(self.dtype)
 
     def forward(self, r_x, r_x_hat):
-        return torch.nn.ReLU(1 + r_x_hat[0]) + torch.nn.ReLU(1 - r_x[0])
+        return F.relu(1 + r_x_hat[0]) + F.relu(1 - r_x[0])