🔧 Increase ViT size and add decoder bounding to cnn_vit_cnn

Author: jemrobinson

icenet_mp/config/model/03_cnn_vit_cnn.yaml

@@ -19,9 +19,13 @@ encoders:
 
 processor:
   _target_: icenet_mp.models.processors.VitProcessor
-  patch_size: 12  # 108 / 12 = 9 patches per dimension
+  depth: 6       # number of transformer blocks
+  dropout: 0.1   # dropout rate
+  emb_dim: 256   # embedding dimension
+  mlp_dim: 1024  # dimension of the MLP in the transformer block
+  patch_size: 6  # 108 / 6 = 18 patches per dimension
 
 decoder:
   _target_: icenet_mp.models.decoders.CNNDecoder
-  bounded: false
+  bounded: true
   n_layers: 2