Skip to content

Commit 0676c07

Browse files
authored
Fix import in minGPT example (#1169)
1 parent 92790cd commit 0676c07

File tree

1 file changed

+1
-0
lines changed
  • distributed/minGPT-ddp/mingpt

1 file changed

+1
-0
lines changed

distributed/minGPT-ddp/mingpt/main.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
1+
import os
12
import torch
23
from torch.utils.data import random_split
34
from torch.distributed import init_process_group, destroy_process_group

0 commit comments

Comments
 (0)