pytorch/examples

DDP training question

Open

#1.096 aberto em 25 de nov. de 2022

Ver no GitHub
 (2 comments) (0 reactions) (1 assignee)Python (21.634 stars) (9.429 forks)batch import
distributedhelp wanted

Description

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?

Guia do colaborador