Fix TypeError on CPU

tsurumeso · Dec 29, 2019 · 5804c6d · 5804c6d
1 parent 47b7249
commit 5804c6d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lib/loss/clipped_weighted_huber_loss.py b/lib/loss/clipped_weighted_huber_loss.py
@@ -34,7 +34,7 @@ def forward(self, inputs):
         xp.square(abs_diff, out=abs_diff)
         y = (y - abs_diff) * 0.5
 
-        return y.mean(),
+        return xp.array(y.mean(), dtype=y.dtype),
 
     def backward(self, inputs, grad_outputs):
         xp = cuda.get_array_module(*inputs)