pytorch/examples

DDP training question

Open

Aperta il 25 nov 2022

Vedi su GitHub
 (2 commenti) (0 reazioni) (1 assegnatario)Python (21.634 star) (9429 fork)batch import
distributedhelp wanted

Descrizione

Hi, I'm using the tutorial https://github.com/pytorch/tutorials/blob/master/intermediate_source/ddp_tutorial.rst for DDP train,using 4 gpus in myself code, reference Basic Use Case. But when I finished the modification, it was stuck during run the demo,meanwhile,video memory has been occupied.Could you help me?

Guida contributor