add cache

Signed-off-by: Can-Zhao <volcanofly@gmail.com>
Project-MONAI · Dec 15, 2023 · 09427c3 · 09427c3
1 parent 3618cc8
commit 09427c3
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 6 deletions.
diff --git a/models/brats_mri_generative_diffusion/configs/train_autoencoder.json b/models/brats_mri_generative_diffusion/configs/train_autoencoder.json
@@ -155,9 +155,10 @@
             "transforms": "$@preprocessing_transforms + @train#crop_transforms + @final_transforms"
         },
         "dataset": {
-            "_target_": "Dataset",
+            "_target_": "CacheDataset",
             "data": "@train_datalist",
-            "transform": "@train#preprocessing"
+            "transform": "@train#preprocessing",
+            "cache_rate": 1.0
         },
         "dataloader": {
             "_target_": "DataLoader",
@@ -223,9 +224,10 @@
             "transforms": "$@preprocessing_transforms + @validate#crop_transforms + @final_transforms"
         },
         "dataset": {
-            "_target_": "Dataset",
+            "_target_": "CacheDataset",
             "data": "@val_datalist",
-            "transform": "@validate#preprocessing"
+            "transform": "@validate#preprocessing",
+            "cache_rate": 1.0
         },
         "dataloader": {
             "_target_": "DataLoader",

diff --git a/models/brats_mri_generative_diffusion/configs/train_diffusion.json b/models/brats_mri_generative_diffusion/configs/train_diffusion.json
@@ -85,9 +85,10 @@
             "transforms": "$@preprocessing_transforms + @train#crop_transforms + @final_transforms"
         },
         "dataset": {
-            "_target_": "Dataset",
+            "_target_": "CacheDataset",
             "data": "@train_datalist",
-            "transform": "@train#preprocessing"
+            "transform": "@train#preprocessing",
+            "cache_rate": 1.0
         },
         "dataloader": {
             "_target_": "DataLoader",