update

train.py

@@ -559,7 +559,7 @@ if __name__ == '__main__':
   local_rank = 0
   if params['world_size'] > 1:
     dist.init_process_group(backend='nccl',
-                            init_method='env://', timeout=timedelta(seconds=10))
+                            init_method='env://', timeout=timedelta(seconds=120))
     local_rank = int(os.environ["LOCAL_RANK"])
     args.gpu = local_rank
     world_rank = dist.get_rank()