How to train on multiple Gpus

Yzmblog / MonoHuman

MonoHuman: Animatable Human Neural Field from Monocular Video (CVPR 2023)

129 stars 9 forks source link

How to train on multiple Gpus #20

Open TShoreWind opened 3 weeks ago

TShoreWind commented 3 weeks ago

@Yzmblog Thank you for your excellent work

Yzmblog commented 3 weeks ago

Hi, thanks for your interests. You can use 'python -m torch.distributed.launch --nproc_per_node 4 train.py --cfg configs/monohuman/zju_mocap/377/377.yaml --ddp resume False' described in the readme.