monai
medical
katielink commited on
Commit
c2efa82
·
1 Parent(s): 46c704b

fix mgpu finalize issue

Browse files
configs/metadata.json CHANGED
@@ -1,7 +1,8 @@
1
  {
2
  "schema": "https://github.com/Project-MONAI/MONAI-extra-test-data/releases/download/0.8.1/meta_schema_20220324.json",
3
- "version": "0.4.5",
4
  "changelog": {
 
5
  "0.4.5": "enable deterministic training",
6
  "0.4.4": "add the command of executing inference with TensorRT models",
7
  "0.4.3": "fix figure and weights inconsistent error",
 
1
  {
2
  "schema": "https://github.com/Project-MONAI/MONAI-extra-test-data/releases/download/0.8.1/meta_schema_20220324.json",
3
+ "version": "0.4.6",
4
  "changelog": {
5
+ "0.4.6": "fix mgpu finalize issue",
6
  "0.4.5": "enable deterministic training",
7
  "0.4.4": "add the command of executing inference with TensorRT models",
8
  "0.4.3": "fix figure and weights inconsistent error",
configs/multi_gpu_evaluate.json CHANGED
@@ -26,6 +26,6 @@
26
  "$@validate#evaluator.run()"
27
  ],
28
  "finalize": [
29
- "$dist.destroy_process_group()"
30
  ]
31
  }
 
26
  "$@validate#evaluator.run()"
27
  ],
28
  "finalize": [
29
+ "$dist.is_initialized() and dist.destroy_process_group()"
30
  ]
31
  }
configs/multi_gpu_train.json CHANGED
@@ -37,6 +37,6 @@
37
  "$@train#trainer.run()"
38
  ],
39
  "finalize": [
40
- "$dist.destroy_process_group()"
41
  ]
42
  }
 
37
  "$@train#trainer.run()"
38
  ],
39
  "finalize": [
40
+ "$dist.is_initialized() and dist.destroy_process_group()"
41
  ]
42
  }