From 42846fb0679d8bf3131fc0570f8285375a96aa78 Mon Sep 17 00:00:00 2001
From: jimmy wang <39302660+b05902062@users.noreply.github.com>
Date: Tue, 20 Aug 2024 21:16:44 +0800
Subject: [PATCH] fix b1c6nbt model invalid mid_num_channels setting.

---
 python/modelconfigs.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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,