- another python 3.7 annoyance

inferno-pytorch · Oct 7, 2018 · 80b6dea · 80b6dea
1 parent 0934395
commit 80b6dea
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/inferno/extensions/layers/device.py b/inferno/extensions/layers/device.py
@@ -33,7 +33,8 @@ def __init__(self, target_device, device_ordinal=None, asynchronous=False):
 
     def forward(self, *inputs):
         if self.target_device == 'cuda':
-            transferred = tuple(input_.cuda(device_id=self.device_ordinal, async=self.asynchronous)
+            # FIXME Removed async support for compatibility with python 3.7.
+            transferred = tuple(input_.cuda(device_id=self.device_ordinal)
                                 for input_ in inputs)
         elif self.target_device == 'cpu':
             transferred = tuple(input_.cpu() for input_ in inputs)