Skip to content

Commit 180d265

Browse files
author
Donglai Wei
committed
update monai2d_worm
1 parent fe1d60f commit 180d265

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

tutorials/monai2d_worm.yaml

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -24,9 +24,9 @@ description: 2D worm segmentation using MONAI Residual UNet with paired data tra
2424
system:
2525
training:
2626
num_gpus: 1 # Single GPU
27-
num_cpus: 1 # Increase for better data loading
28-
num_workers: 1 # Parallel data loading (2D slices are lighter)
29-
batch_size: 8 # Higher batch size for 2D (vs 4 for 3D)
27+
num_cpus: 4 # Increase for better data loading
28+
num_workers: 4 # Parallel data loading (2D slices are lighter)
29+
batch_size: 32 # Higher batch size for 2D (vs 4 for 3D)
3030
inference:
3131
num_gpus: 1
3232
num_cpus: 4
@@ -59,8 +59,8 @@ model:
5959
# Data - Using automatic 80/20 train/val split (DeepEM-style)
6060
data:
6161
# Volume configuration
62-
train_image: /projects/weilab/shenb/PyTC/datasets/Dataset001_worm_image96/imagesTr/Image96_00001_0000.tif
63-
train_label: /projects/weilab/shenb/PyTC/datasets/Dataset001_worm_image96/labelsTr/Image96_00001.tif
62+
train_image: /projects/weilab/shenb/PyTC/datasets/Dataset001_worm_image96/imagesTr/*.tif
63+
train_label: /projects/weilab/shenb/PyTC/datasets/Dataset001_worm_image96/labelsTr/*.tif
6464
train_resolution: [5, 5] # Lucchi EM: 5nm isotropic resolution
6565
use_preloaded_cache: true # Load volumes into memory for fast training
6666

0 commit comments

Comments
 (0)