Update imports (pytorch#1155)

Update main.py
mpc001 · May 25, 2023 · 55c663f · 55c663f
1 parent 7b7c708
commit 55c663f
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/distributed/minGPT-ddp/mingpt/main.py b/distributed/minGPT-ddp/mingpt/main.py
@@ -1,10 +1,12 @@
+import torch
+from torch.utils.data import random_split
+from torch.distributed import init_process_group, destroy_process_group
 from model import GPT, GPTConfig, OptimizerConfig, create_optimizer
 from trainer import Trainer, TrainerConfig
 from char_dataset import CharDataset, DataConfig
-from torch.utils.data import random_split
 from omegaconf import DictConfig
 import hydra
-from torch.distributed import init_process_group, destroy_process_group
+
 
 def ddp_setup():
     init_process_group(backend="nccl")