train script using torchrun for torch version >= 2.0.0

OFA-Sys / Chinese-CLIP

Chinese version of CLIP which achieves Chinese cross-modal retrieval and representation generation.

MIT License

4.58k stars 473 forks source link

train script using torchrun for torch version >= 2.0.0 #349

Open SuperAZHE opened 3 months ago

SuperAZHE commented 3 months ago

torch 2.0.0 use torchrun。 The module torch.distributed.launch is deprecated and will be removed in future. Use torchrun. Note that --use-env is set by default in torchrun.