Skip to content

GitLab

Unverified Commit 5a39c5e3 authored by mcarilli's avatar mcarilli Committed by GitHub
Browse files

Update distributed.py

parent 034b8f02
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
apex/parallel/distributed.py
View file @ 5a39c5e3
...@@ -4,7 +4,7 @@ import torch.distributed as dist ...@@ -4,7 +4,7 @@ import torch.distributed as dist
from torch.nn.modules import Module from torch.nn.modules import Module
from torch.autograd import Variable from torch.autograd import Variable
from collections import OrderedDict from collections import OrderedDict
import copy
def flat_dist_call(tensors, call, extra_args=None): def flat_dist_call(tensors, call, extra_args=None):
flat_dist_call.warn_on_half = True flat_dist_call.warn_on_half = True
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment