pytorch/examples

DDP training question

Ouverte

#1 096 ouverte le 25 nov. 2022

 (2 commentaires) (0 réaction) (1 personne assignée)Python (9 429 forks)batch import
distributedhelp wanted

Métriques du dépôt

Stars
 (21 634 étoiles)
Métriques de merge PR
 (Métriques PR en attente)

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