Can-Zhao commited on
Commit
84e1e7c
·
1 Parent(s): 569edd6

add network def

Browse files

Signed-off-by: Can-Zhao <[email protected]>

Files changed (1) hide show
  1. configs/config_network_ddpm.json +6 -2
configs/config_network_ddpm.json CHANGED
@@ -63,7 +63,9 @@
63
  "use_flash_attention": true,
64
  "include_top_region_index_input": "@include_body_region",
65
  "include_bottom_region_index_input": "@include_body_region",
66
- "include_spacing_input": true
 
 
67
  },
68
  "controlnet_def": {
69
  "_target_": "monai.apps.generation.maisi.networks.controlnet_maisi.ControlNetMaisi",
@@ -90,7 +92,9 @@
90
  "num_res_blocks": 2,
91
  "use_flash_attention": true,
92
  "conditioning_embedding_in_channels": 8,
93
- "conditioning_embedding_num_channels": [8, 32, 64]
 
 
94
  },
95
  "mask_generation_autoencoder_def": {
96
  "_target_": "monai.apps.generation.maisi.networks.autoencoderkl_maisi.AutoencoderKlMaisi",
 
63
  "use_flash_attention": true,
64
  "include_top_region_index_input": "@include_body_region",
65
  "include_bottom_region_index_input": "@include_body_region",
66
+ "include_spacing_input": true,
67
+ "include_fc": true,
68
+ "resblock_updown": false
69
  },
70
  "controlnet_def": {
71
  "_target_": "monai.apps.generation.maisi.networks.controlnet_maisi.ControlNetMaisi",
 
92
  "num_res_blocks": 2,
93
  "use_flash_attention": true,
94
  "conditioning_embedding_in_channels": 8,
95
+ "conditioning_embedding_num_channels": [8, 32, 64],
96
+ "include_fc": true,
97
+ "resblock_updown": false
98
  },
99
  "mask_generation_autoencoder_def": {
100
  "_target_": "monai.apps.generation.maisi.networks.autoencoderkl_maisi.AutoencoderKlMaisi",