monai
medical
katielink commited on
Commit
64fd9ca
1 Parent(s): 76df71c

enable deterministic training

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.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.0rc3",
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()"