search

InfluenceFunctional / MXtalTools

BSD 3-Clause "New" or "Revised" License
7 stars 1 forks source link

Implement and test multi-GPU training #74

Open InfluenceFunctional opened 1 year ago

InfluenceFunctional commented 1 year ago

Discriminator training is slow - could possibly be sped up with nn.distributeddataparallel

InfluenceFunctional commented 1 year ago

nn.dataparallel with a the pyg parallel loader should be simple to implement

nn.distributeddataparallel is generally more highly recommended, but is more work (I think) to implement