multi-gpu use

[Xcrid]

Hi,

I just wanted to know if this implementation would work well in a DataParrallel model and if there was some adaptation to do ?

Thanks

[ndrplz]

Hi @Xcrid , I can't really answer as I'm not much into pytorch data parallelism and lately I'm working in TF

I'm keeping the issue open for someone else to contribute. If you find the answer youself, please write it here s.t. it can be helpful to others

Best, A