nv-tlabs / GET3D

Other
4.2k stars 376 forks source link

Typo in train_3d.py #121

Closed kdha0727 closed 1 year ago

kdha0727 commented 1 year ago
@@ -30,7 +30,7 @@ def subprocess_fn(rank, c, temp_dir):
         if os.name == 'nt':
             init_method = 'file:///' + init_file.replace('\\', '/')
             torch.distributed.init_process_group(
-                backend='gloo', icfgnit_method=init_method, rank=rank, world_size=c.num_gpus)
+                backend='gloo', init_method=init_method, rank=rank, world_size=c.num_gpus)
         else:
             init_method = f'file://{init_file}'
             torch.distributed.init_process_group(

keyword name of torch.distributed.init_process_group should be modified.

SteveJunGao commented 1 year ago

Thanks for letting us know! I just updated the function