my bad. torch.nn.parallel.DistributedDataParallel
, not torch.nn.DistributedDataParallel
1 Like
my bad. torch.nn.parallel.DistributedDataParallel
, not torch.nn.DistributedDataParallel