pytorch/examples

DDP training question

Aberta

#1.096 aberto em 25 de nov. de 2022

 (2 comentários) (0 reação) (1 responsável)Python (9.429 forks)batch import
distributedhelp wanted

Métricas do repositório

Stars
 (21.634 estrelas)
Métricas de merge de PR
 (Métricas PR pendentes)

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