Changes model dims

brlp_lite.py

@@ -72,10 +72,11 @@ INPUT_SHAPE_1mm = (182, 218, 182)
 INPUT_SHAPE_1p5mm = (122, 146, 122)   
 
 # Adjusting the dimensions to be divisible by 8 (2^3 where 3 are the downsampling layers of the AE)
-INPUT_SHAPE_AE = (120, 144, 120)   
+#INPUT_SHAPE_AE = (120, 144, 120)
+INPUT_SHAPE_AE = (80, 96, 80)
 
 # Latent shape of the autoencoder 
-LATENT_SHAPE_AE = (3, 15, 18, 15)   
+LATENT_SHAPE_AE = (1, 10, 12, 10)   
 
 
 def load_if(checkpoints_path: Optional[str], network: nn.Module) -> nn.Module:
@@ -111,7 +112,7 @@ def init_autoencoder(checkpoints_path: Optional[str] = None) -> nn.Module:
     autoencoder = AutoencoderKL(spatial_dims=3, 
                                 in_channels=1, 
                                 out_channels=1, 
-                                latent_channels=3,
+                                latent_channels=1, #3,
                                 num_channels=(64, 128, 128, 128),
                                 num_res_blocks=2, 
                                 norm_num_groups=32,

runs/Jan31_14-52-36_SOM-YT7DYVX-DT/events.out.tfevents.1738363956.SOM-YT7DYVX-DT.46314.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f7fa2b1c9d0b68193dc5deed1e0f6dc775a2b6a7ae04cd17e0c4268ee4460451
+size 184408