Thank you for your code! I can run it successfully with one GPU. But when I ran the code with multiple GPUs, I got the following error:
Traceback (most recent call last):
File "/opt/conda/lib/python3.8/site-packages/torch/multiprocessing/spawn.py", line 59, in _wrap
fn(i, *args)
File "/3dod2/Workspace/irem_code_release/train.py", line 690, in main_single
sync_model(model)
File "/3dod2/Workspace/irem_code_release/train.py", line 305, in sync_model
for model in model:
TypeError: 'EBM' object is not iterable
Thank you for your code! I can run it successfully with one GPU. But when I ran the code with multiple GPUs, I got the following error:
Any suggestions? Thanks!