From 09427c365eaac6d4c7589e4764cdcb10cfe8fc11 Mon Sep 17 00:00:00 2001 From: Can-Zhao Date: Thu, 14 Dec 2023 17:33:34 -0800 Subject: [PATCH] add cache Signed-off-by: Can-Zhao --- .../configs/train_autoencoder.json | 10 ++++++---- .../configs/train_diffusion.json | 5 +++-- 2 files changed, 9 insertions(+), 6 deletions(-) diff --git a/models/brats_mri_generative_diffusion/configs/train_autoencoder.json b/models/brats_mri_generative_diffusion/configs/train_autoencoder.json index e4bf0669..3086aa44 100644 --- 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 index 5bb55a7c..e699fa50 100644 --- 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",