Update loss.py

fix incorrect loss function
meetps · Mar 6, 2019 · 801fb20 · 801fb20
1 parent bc523a7
commit 801fb20
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ptsemseg/loss/loss.py b/ptsemseg/loss/loss.py
@@ -27,7 +27,7 @@ def multi_scale_cross_entropy2d(input, target, weight=None, size_average=True, s
         n_inp = len(input)
         scale = 0.4
         scale_weight = torch.pow(scale * torch.ones(n_inp), torch.arange(n_inp).float()).to(
-            input.device
+            target.device
         )
 
     loss = 0.0