From 5e05252801f6d145e94717207419cb348fe9ea64 Mon Sep 17 00:00:00 2001
From: bendboaz <boaz.ben-dov@campus.technion.ac.il>
Date: Sun, 5 Apr 2020 11:42:56 +0300
Subject: [PATCH] Fixed device conversion method (#887)

Co-authored-by: Boaz <bendboaz@gmail.com>
---
 ignite/utils.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ignite/utils.py b/ignite/utils.py
index 291294d52d4..59a8bc12439 100644
--- 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)