From de9e7f19b8b60645b2d94d210b307cca8d06e926 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Gordon=20Guocheng=20Qian=20=E9=92=B1=E5=9B=BD=E6=88=90?= Date: Wed, 21 Dec 2022 10:43:14 -0800 Subject: [PATCH] Update main.py --- examples/segmentation/main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/segmentation/main.py b/examples/segmentation/main.py index b28a894..bf32cec 100644 --- 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)