Need a multi-GPU version

vincent-leguen / PhyDNet

Code for our CVPR 2020 paper "Disentangling Physical Dynamics from Unknown Factors for UnsupervisedVideo Prediction"

MIT License

176 stars 50 forks source link

Open ChangdeDu opened 4 years ago

ChangdeDu commented 4 years ago

Currently, it does not work with nn.DataParallel(encoder). Need a multi-GPU version.