Skip to content

Commit f085a28

Browse files
Can-ZhaoCan-Zhao
committed
add modality as input to config
Signed-off-by: Can-Zhao <[email protected]>
1 parent 151177d commit f085a28

11 files changed

+122
-120
lines changed

‎generation/maisi/configs/config_infer_16g_256x256x128.json‎

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
"body_region": ["abdomen"],
44
"anatomy_list": ["liver","hepatic tumor"],
55
"controllable_anatomy_size": [],
6-
"num_inference_steps": 1000,
6+
"num_inference_steps": 30,
77
"mask_generation_num_inference_steps": 1000,
88
"output_size": [
99
256,
@@ -19,5 +19,6 @@
1919
],
2020
"autoencoder_sliding_window_infer_size": [96,96,96],
2121
"autoencoder_sliding_window_infer_overlap": 0.25,
22-
"autoencoder_tp_num_splits": 2
22+
"autoencoder_tp_num_splits": 2,
23+
"modality": 1
2324
}

‎generation/maisi/configs/config_infer_16g_256x256x256.json‎

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
"body_region": ["abdomen"],
44
"anatomy_list": ["liver","hepatic tumor"],
55
"controllable_anatomy_size": [],
6-
"num_inference_steps": 1000,
6+
"num_inference_steps": 30,
77
"mask_generation_num_inference_steps": 1000,
88
"output_size": [
99
256,
@@ -19,5 +19,6 @@
1919
],
2020
"autoencoder_sliding_window_infer_size": [48,48,64],
2121
"autoencoder_sliding_window_infer_overlap": 0.25,
22-
"autoencoder_tp_num_splits": 2
22+
"autoencoder_tp_num_splits": 2,
23+
"modality": 1
2324
}

‎generation/maisi/configs/config_infer_16g_512x512x128.json‎

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
"body_region": ["abdomen"],
44
"anatomy_list": ["liver","hepatic tumor"],
55
"controllable_anatomy_size": [],
6-
"num_inference_steps": 1000,
6+
"num_inference_steps": 30,
77
"mask_generation_num_inference_steps": 1000,
88
"output_size": [
99
512,
@@ -19,5 +19,6 @@
1919
],
2020
"autoencoder_sliding_window_infer_size": [64,64,32],
2121
"autoencoder_sliding_window_infer_overlap": 0.25,
22-
"autoencoder_tp_num_splits": 1
22+
"autoencoder_tp_num_splits": 1,
23+
"modality": 1
2324
}

‎generation/maisi/configs/config_infer_24g_256x256x256.json‎

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
"body_region": ["abdomen"],
44
"anatomy_list": ["liver","hepatic tumor"],
55
"controllable_anatomy_size": [],
6-
"num_inference_steps": 1000,
6+
"num_inference_steps": 30,
77
"mask_generation_num_inference_steps": 1000,
88
"output_size": [
99
256,
@@ -19,5 +19,6 @@
1919
],
2020
"autoencoder_sliding_window_infer_size": [64,64,64],
2121
"autoencoder_sliding_window_infer_overlap": 0.25,
22-
"autoencoder_tp_num_splits": 4
22+
"autoencoder_tp_num_splits": 4,
23+
"modality": 1
2324
}

‎generation/maisi/configs/config_infer_24g_512x512x128.json‎

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
"body_region": ["abdomen"],
44
"anatomy_list": ["liver","hepatic tumor"],
55
"controllable_anatomy_size": [],
6-
"num_inference_steps": 1000,
6+
"num_inference_steps": 30,
77
"mask_generation_num_inference_steps": 1000,
88
"output_size": [
99
512,
@@ -19,5 +19,6 @@
1919
],
2020
"autoencoder_sliding_window_infer_size": [80,80,32],
2121
"autoencoder_sliding_window_infer_overlap": 0.25,
22-
"autoencoder_tp_num_splits": 1
22+
"autoencoder_tp_num_splits": 1,
23+
"modality": 1
2324
}

‎generation/maisi/configs/config_infer_24g_512x512x512.json‎

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
"body_region": ["abdomen"],
44
"anatomy_list": ["liver","hepatic tumor"],
55
"controllable_anatomy_size": [],
6-
"num_inference_steps": 1000,
6+
"num_inference_steps": 30,
77
"mask_generation_num_inference_steps": 1000,
88
"output_size": [
99
512,
@@ -19,5 +19,6 @@
1919
],
2020
"autoencoder_sliding_window_infer_size": [64,64,48],
2121
"autoencoder_sliding_window_infer_overlap": 0.25,
22-
"autoencoder_tp_num_splits": 2
22+
"autoencoder_tp_num_splits": 2,
23+
"modality": 1
2324
}

‎generation/maisi/configs/config_infer_32g_512x512x512.json‎

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
"body_region": ["abdomen"],
44
"anatomy_list": ["liver","hepatic tumor"],
55
"controllable_anatomy_size": [],
6-
"num_inference_steps": 1000,
6+
"num_inference_steps": 30,
77
"mask_generation_num_inference_steps": 1000,
88
"output_size": [
99
512,
@@ -19,5 +19,6 @@
1919
],
2020
"autoencoder_sliding_window_infer_size": [64,64,64],
2121
"autoencoder_sliding_window_infer_overlap": 0.25,
22-
"autoencoder_tp_num_splits": 2
22+
"autoencoder_tp_num_splits": 2,
23+
"modality": 1
2324
}

‎generation/maisi/configs/config_infer_80g_512x512x128.json‎

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
"body_region": ["abdomen"],
44
"anatomy_list": ["liver","hepatic tumor"],
55
"controllable_anatomy_size": [],
6-
"num_inference_steps": 1000,
6+
"num_inference_steps": 30,
77
"mask_generation_num_inference_steps": 1000,
88
"output_size": [
99
512,
@@ -19,5 +19,6 @@
1919
],
2020
"autoencoder_sliding_window_infer_size": [128,128,32],
2121
"autoencoder_sliding_window_infer_overlap": 0.25,
22-
"autoencoder_tp_num_splits": 4
22+
"autoencoder_tp_num_splits": 4,
23+
"modality": 1
2324
}

‎generation/maisi/configs/config_infer_80g_512x512x512.json‎

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
"body_region": ["abdomen"],
44
"anatomy_list": ["liver","hepatic tumor"],
55
"controllable_anatomy_size": [],
6-
"num_inference_steps": 1000,
6+
"num_inference_steps": 30,
77
"mask_generation_num_inference_steps": 1000,
88
"output_size": [
99
512,
@@ -19,5 +19,6 @@
1919
],
2020
"autoencoder_sliding_window_infer_size": [80,80,80],
2121
"autoencoder_sliding_window_infer_overlap": 0.25,
22-
"autoencoder_tp_num_splits": 2
22+
"autoencoder_tp_num_splits": 2,
23+
"modality": 1
2324
}

‎generation/maisi/configs/config_infer_80g_512x512x768.json‎

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
"body_region": ["abdomen"],
44
"anatomy_list": ["liver","hepatic tumor"],
55
"controllable_anatomy_size": [],
6-
"num_inference_steps": 1000,
6+
"num_inference_steps": 30,
77
"mask_generation_num_inference_steps": 1000,
88
"output_size": [
99
512,
@@ -19,5 +19,6 @@
1919
],
2020
"autoencoder_sliding_window_infer_size": [80,80,112],
2121
"autoencoder_sliding_window_infer_overlap": 0.25,
22-
"autoencoder_tp_num_splits": 4
22+
"autoencoder_tp_num_splits": 4,
23+
"modality": 1
2324
}

0 commit comments

Comments
 (0)