open-mmlab / mmgeneration

MMGeneration is a powerful toolkit for generative models, based on PyTorch and MMCV.
https://mmgeneration.readthedocs.io/en/latest/
Apache License 2.0
1.91k stars 232 forks source link

in launch_agent raise ChildFailedError( torch.distributed.elastic.multiprocessing.errors.ChildFailedError: #515

Open Cuppinono opened 1 year ago

Cuppinono commented 1 year ago

bash tools/dist_train.sh configs/styleganv2/stylegan2_c2_ffhq_256_b4x8_800k_strawberry.py 1 --work-dir work_dirs/experiments/stylegan2_strawberry

error:train.py: error: unrecognized arguments: --local-rank=0,ERROR:torch.distributed.elastic.multiprocessing.api:failed (exitcode: 2) local_rank: 0 (pid: 27820) of binary: /home/dell/anaconda3/envs/zcc/bin/python