Adjusting image sizes to prevent out of bounds error
@@ -15,14 +15,14 @@ net_name = vnet
# preprocessing threads parameters
queue_length = 8
-num_threads = 1
+num_threads = 2
# network parameters (controlling memory consumption)
spatial_rank = 3
batch_size = 1
-image_size = 128
-label_size = 128
-w_map_size = 128
+image_size = 64
+label_size = 64
+w_map_size = 64
num_classes = 2
# affected by network's receptive field
@@ -47,7 +47,7 @@ w_map_interp_order = 3
# ** training only ** system parameters
num_gpus = 1
-sample_per_volume = 1
+sample_per_volume = 4
# training augmentation
rotation = True
@@ -63,8 +63,8 @@ decay = 1e-7
loss_type = Dice
reg_type = L2
starting_iter = 0
-save_every_n = 500
-max_iter = 1000
+save_every_n = 12500
+max_iter = 25000
# ** inference only ** system parameters
border = 5
