pytorch/examples

DDP training question

Open

#1 096 ouverte le 25 nov. 2022

Voir sur GitHub
 (2 commentaires) (0 réactions) (1 assigné)Python (9 429 forks)batch import
distributedhelp wanted

Métriques du dépôt

Stars
 (21 634 stars)
Métriques de merge PR
 (Aucune PR mergée en 30 j)

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?

Guide contributeur