enable deterministic training
Browse files- configs/metadata.json +3 -2
- configs/multi_gpu_train.json +1 -2
- configs/train.json +1 -2
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.
|
| 4 |
"changelog": {
|
|
|
|
| 5 |
"0.4.4": "update numbers",
|
| 6 |
"0.4.3": "adapt to BundleWorkflow interface",
|
| 7 |
"0.4.2": "fix train params of use_checkpoint",
|
|
@@ -21,7 +22,7 @@
|
|
| 21 |
"0.1.0": "complete the model package",
|
| 22 |
"0.0.1": "initialize the model package structure"
|
| 23 |
},
|
| 24 |
-
"monai_version": "1.2.
|
| 25 |
"pytorch_version": "1.13.1",
|
| 26 |
"numpy_version": "1.22.2",
|
| 27 |
"optional_packages_version": {
|
|
|
|
| 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": "update numbers",
|
| 7 |
"0.4.3": "adapt to BundleWorkflow interface",
|
| 8 |
"0.4.2": "fix train params of use_checkpoint",
|
|
|
|
| 22 |
"0.1.0": "complete the model package",
|
| 23 |
"0.0.1": "initialize the model package structure"
|
| 24 |
},
|
| 25 |
+
"monai_version": "1.2.0rc4",
|
| 26 |
"pytorch_version": "1.13.1",
|
| 27 |
"numpy_version": "1.22.2",
|
| 28 |
"optional_packages_version": {
|
configs/multi_gpu_train.json
CHANGED
|
@@ -28,8 +28,7 @@
|
|
| 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)"
|
| 33 |
],
|
| 34 |
"run": [
|
| 35 |
"$@train#trainer.run()"
|
|
|
|
| 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 |
],
|
| 33 |
"run": [
|
| 34 |
"$@train#trainer.run()"
|
configs/train.json
CHANGED
|
@@ -319,8 +319,7 @@
|
|
| 319 |
}
|
| 320 |
},
|
| 321 |
"initialize": [
|
| 322 |
-
"$monai.utils.set_determinism(seed=123)"
|
| 323 |
-
"$setattr(torch.backends.cudnn, 'benchmark', True)"
|
| 324 |
],
|
| 325 |
"run": [
|
| 326 |
"$@train#trainer.run()"
|
|
|
|
| 319 |
}
|
| 320 |
},
|
| 321 |
"initialize": [
|
| 322 |
+
"$monai.utils.set_determinism(seed=123)"
|
|
|
|
| 323 |
],
|
| 324 |
"run": [
|
| 325 |
"$@train#trainer.run()"
|