async --> non_blocking for Python 3.7 compatibility

Fixes CSAILVision#7
cclauss · Mar 7, 2019 · a104f3b · a104f3b
1 parent bb947de
commit a104f3b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lib/nn/parallel/data_parallel.py b/lib/nn/parallel/data_parallel.py
@@ -13,7 +13,7 @@ def async_copy_to(obj, dev, main_stream=None):
     if torch.is_tensor(obj):
         obj = Variable(obj)
     if isinstance(obj, Variable):
-        v = obj.cuda(dev, async=True)
+        v = obj.cuda(dev, non_blocking=True)
         if main_stream is not None:
             v.data.record_stream(main_stream)
         return v