Fix for train.py

codes/train.py

@@ -293,7 +293,7 @@ class Trainer:
 
 if __name__ == '__main__':
     parser = argparse.ArgumentParser()
-    parser.add_argument('-opt', type=str, help='Path to option YAML file.', default='../options/train_xx_faces_glean.yml')
+    parser.add_argument('-opt', type=str, help='Path to option YAML file.', default='../options/train_faces_styled_sr.yml')
     parser.add_argument('--launcher', choices=['none', 'pytorch'], default='none', help='job launcher')
     parser.add_argument('--local_rank', type=int, default=0)
     args = parser.parse_args()
@@ -306,7 +306,7 @@ if __name__ == '__main__':
             print('export CUDA_VISIBLE_DEVICES=' + gpu_list)
     trainer = Trainer()
 
-#### distributed training settings
+    #### distributed training settings
     if args.launcher == 'none':  # disabled distributed training
         opt['dist'] = False
         trainer.rank = -1
@@ -315,7 +315,7 @@ if __name__ == '__main__':
         print('Disabled distributed training.')
     else:
         opt['dist'] = True
-        init_dist('nccl', opt)
+        init_dist('nccl')
         trainer.world_size = torch.distributed.get_world_size()
         trainer.rank = torch.distributed.get_rank()