Skip to content

Commit fd86def

Browse files
update error explanation in parameters fold in tutorials train_controlnet.ipynb (#1974)
Fixes # . Description The current documentation incorrectly states that "fold 0 is used for training" in the prepare_maisi_controlnet_json_dataloader function. In reality, the function uses the fold parameter to determine the validation set based on matching fold indices. Specifically, if a dataset item's fold value matches the fold value specified in config_maisi_controlnet_train.json, it is assigned to the validation set; otherwise, it is used for training. Changes Updated Documentation: Revised the description of the fold parameter to accurately reflect its functionality. Clarified that in the current parameter settings, fold 0 is designated for validation and fold 1 is designated for training. --------- Signed-off-by: MaybeRichard <lisicheng168@gmail.com> Co-authored-by: Eric Kerfoot <17726042+ericspod@users.noreply.github.com>
1 parent 84c0a07 commit fd86def

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

generation/maisi/maisi_train_controlnet_tutorial.ipynb

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -189,7 +189,7 @@
189189
},
190190
{
191191
"cell_type": "code",
192-
"execution_count": 4,
192+
"execution_count": null,
193193
"id": "fc32a7fe",
194194
"metadata": {},
195195
"outputs": [],
@@ -201,7 +201,9 @@
201201
" \"image\": \"tr_image_001_emb.nii.gz\", # relative path to the image embedding file\n",
202202
" # relative path to the combined label (pseudo whole-body segmentation mask + ROI mask) file\n",
203203
" \"label\": \"tr_label_001.nii.gz\",\n",
204-
" \"fold\": 0, # fold index for cross validation, fold 0 is used for training\n",
204+
" \"fold\": 0, # fold index for cross validation. If the dataset item's fold value is equal to the fold value\n",
205+
" # in config_maisi_controlnet_train.json, then it is used for validation. Otherwise, it is used for training.\n",
206+
" # In the current parameters, fold 0 is for validation.\n",
205207
" \"dim\": sim_dim, # the dimension of image\n",
206208
" \"spacing\": [1.5, 1.5, 1.5], # the spacing of image\n",
207209
" },\n",

0 commit comments

Comments
 (0)