Bob
Bob
> > 还没有 > > 我也没有，可以讨论讨论 请问一下你们现在解决了吗？
Hi, is this repository compatible with the RTX 50-series GPUs? When I run the code with a single GPU (setting gpus=[0] or gpus=[1]), it works fine. However, when I try...
I found that replacing **torch.nn.parallel.DistributedDataParallel(model)** with **torch.nn.parallel.DataParallel(model)** in the **train.py** file resolved the error in the multi-GPU training script. I hope this information helps others who might encounter the similar...