search

NVlabs / edm

Elucidating the Design Space of Diffusion-Based Generative Models (EDM)
Other
1.42k stars 147 forks source link

params[x] in this process with sizes [x, x] appears not to match sizes of the same param in process 0 #11

Closed forever208 closed 1 year ago

forever208 commented 1 year ago

if people have this issue when training with multiple GPUs, try with python3.8 + PyTorch 1.12.1. I had this error when using Python 3.9

kugwzk commented 1 year ago

How did you solve this problem? @forever208

huanranchen commented 1 year ago

How did you solve this problem? @forever208

try with python3.8 + PyTorch 1.12.1.