adding cvg temp fix (NVIDIA-Merlin#427)

jperez999 · benfred · web-flow · commit e0e3ccaa2333 · 2020-11-10T21:02:02.000-08:00
* adding cvg temp fix

* add tensor check

Co-authored-by: Ben Frederickson &lt;github@benfrederickson.com&gt;
diff --git a/nvtabular/loader/torch.py b/nvtabular/loader/torch.py
@@ -107,6 +107,11 @@ def _LONG_DTYPE(self):
     def _FLOAT32_DTYPE(self):
         return torch.float32
 
+    def _handle_tensors(self, cats, conts, labels):
+        if isinstance(conts, torch.Tensor):
+            conts = conts.clone()
+        return cats, conts, labels
+
 
 class DLDataLoader(torch.utils.data.DataLoader):
     """
