Fixed device conversion method (#887)

Co-authored-by: Boaz <bendboaz@gmail.com>
pytorch · Apr 5, 2020 · 5e05252 · 5e05252
1 parent 3d2be23
commit 5e05252
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ignite/utils.py b/ignite/utils.py
@@ -15,7 +15,7 @@ def convert_tensor(
     """Move tensors to relevant device."""
 
     def _func(tensor: torch.Tensor) -> torch.Tensor:
-        return tensor.to(device=device, non_blocking=non_blocking) if device else tensor
+        return tensor.to(device=device, non_blocking=non_blocking) if device is not None else tensor
 
     return apply_to_tensor(input_, _func)