diff --git a/python/modelconfigs.py b/python/modelconfigs.py
index 7bfe24ac6..e60a8da19 100644
--- a/python/modelconfigs.py
+++ b/python/modelconfigs.py
@@ -77,7 +77,7 @@ def get_num_meta_encoder_input_features(config_or_meta_encoder_version: Union[Mo
     "bnorm_running_avg_momentum": 0.001,
     "initial_conv_1x1": False,
     "trunk_num_channels":6,
-    "mid_num_channels":4,
+    "mid_num_channels":6,
     "gpool_num_channels":4,
     "use_attention_pool":False,
     "num_attention_pool_heads":2,