fix multi-gpu train config typo
Browse files- configs/metadata.json +2 -1
- configs/multi_gpu_train.json +1 -1
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 |
"changelog": {
|
|
|
|
| 5 |
"0.3.9": "adapt to BundleWorkflow interface",
|
| 6 |
"0.3.8": "add name tag",
|
| 7 |
"0.3.7": "restructure readme to match updated template",
|
|
|
|
| 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.0",
|
| 4 |
"changelog": {
|
| 5 |
+
"0.4.0": "fix multi-gpu train config typo",
|
| 6 |
"0.3.9": "adapt to BundleWorkflow interface",
|
| 7 |
"0.3.8": "add name tag",
|
| 8 |
"0.3.7": "restructure readme to match updated template",
|
configs/multi_gpu_train.json
CHANGED
|
@@ -26,7 +26,7 @@
|
|
| 26 |
"validate#evaluator#val_handlers": "$None if dist.get_rank() > 0 else @validate#handlers",
|
| 27 |
"initialize": [
|
| 28 |
"$import torch.distributed as dist",
|
| 29 |
-
"dist.is_initialized() or dist.init_process_group(backend='nccl')",
|
| 30 |
"$torch.cuda.set_device(@device)",
|
| 31 |
"$monai.utils.set_determinism(seed=123)",
|
| 32 |
"$setattr(torch.backends.cudnn, 'benchmark', True)",
|
|
|
|
| 26 |
"validate#evaluator#val_handlers": "$None if dist.get_rank() > 0 else @validate#handlers",
|
| 27 |
"initialize": [
|
| 28 |
"$import torch.distributed as dist",
|
| 29 |
+
"$dist.is_initialized() or dist.init_process_group(backend='nccl')",
|
| 30 |
"$torch.cuda.set_device(@device)",
|
| 31 |
"$monai.utils.set_determinism(seed=123)",
|
| 32 |
"$setattr(torch.backends.cudnn, 'benchmark', True)",
|