lesliejackson / PyTorch-Distributed-Training

Example of PyTorch DistributedDataParallel
59 stars 24 forks source link

fix: no argument parser for 'world_size' #4

Closed Alex-HaochenLi closed 9 months ago

Alex-HaochenLi commented 3 years ago

hello, in your original codes, there is no argument for receiving world_size. However, in the function of _reduceloss, we need world_size as a parameter , and this need to be received by argument parser.