Update main.py

guochengqian · Dec 21, 2022 · de9e7f1 · de9e7f1
1 parent 801ebee
commit de9e7f1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/segmentation/main.py b/examples/segmentation/main.py
@@ -327,7 +327,7 @@ def main(gpu, cfg):
         logging.warning('Testing using multiple GPUs is not allowed for now. Running testing after this training is required.')
     if writer is not None:
         writer.close()
-    dist.destroy_process_group()
+    # dist.destroy_process_group() # comment this line due to https://github.com/guochengqian/PointNeXt/issues/95
     wandb.finish(exit_code=True)