Update imports (#1155)

Update main.py

Author: subramen

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")