search

facebookresearch / stochastic_gradient_push

Stochastic Gradient Push for Distributed Deep Learning
Other
158 stars 37 forks source link

Make code compatible with Pytorch-1.4 #10

Open vtantia opened 4 years ago

vtantia commented 4 years ago

In order to make the code compatible with Pytorch-1.4, use of .data.set_ needs to be replaced as mentioned here. Other uses of .data need to be replaced as well in the code