Update loss.py

zzh8829 · meetps · commit 801fb200547c · 2019-03-06T08:12:20.000-08:00
fix incorrect loss function
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

Original file line number	Diff line number	Diff line change
`@@ -27,7 +27,7 @@ def multi_scale_cross_entropy2d(input, target, weight=None, size_average=True, s`
`27`	`27`	`n_inp = len(input)`
`28`	`28`	`scale = 0.4`
`29`	`29`	`scale_weight = torch.pow(scale * torch.ones(n_inp), torch.arange(n_inp).float()).to(`
`30`		`- input.device`
	`30`	`+ target.device`
`31`	`31`	`)`
`32`	`32`
`33`	`33`	`loss = 0.0`