有关分布式训练的问题

Algolzw / daclip-uir

[ICLR 2024] Controlling Vision-Language Models for Universal Image Restoration. 5th place in the NTIRE 2024 Restore Any Image Model in the Wild Challenge.

https://algolzw.github.io/daclip-uir

MIT License

582 stars 30 forks source link

有关分布式训练的问题 #58

Closed Cyyyyyb closed 4 days ago

Cyyyyyb commented 1 week ago

作者您好，非常感谢您提供了如此优秀的工作的代码。在使用分布式训练的代码时，我遇到了一些问题，想来请求您的帮助。根据主页提示，我输入 python3 -m torch.distributed.launch --nproc_per_node=2 --master_port=4321 train.py -opt=options/train.yml --launcher pytorch 这一行代码，报错：

谢谢！

Cyyyyyb commented 5 days ago

这个问题已经解决：将train.py中的local_rank改为local-rank即可