Single node GPU training example (#333)#352

Merged

cosmicBboy merged 2 commits intomasterfrom pytorch-multi-gpu

Aug 17, 2021

+466-137

Commits on Aug 16, 2021

update pytorch multi-gpu example, incorporate comments @samhita-alla @kumare3
cosmicBboy
committed

Commits on Aug 17, 2021

Apply suggestions from code review

cosmicBboy
and
samhita-alla
committed