Home Explore Help
Sign In
junxiaoyao
/
RT-ODLab
1
0
Fork 0
Files Issues 0 Pull Requests 0 Wiki
Browse Source

set com_dim = 300

yjh0410 2 years ago
parent
b2b2e2e342
commit
0ec56185de
1 changed files with 1 additions and 1 deletions
Unified View Show Diff Stats
  1. 1 1
      config/model_config/rtdetr_config.py

+ 1 - 1
config/model_config/rtdetr_config.py
View File 

@@ -32,7 +32,7 @@ rtdetr_cfg = {
 
         'fpn_norm': 'BN',
 
         'fpn_norm': 'BN',
 
         'fpn_depthwise': False,
 
         'fpn_depthwise': False,
 
         ## ------- Memory Decoder -------
 
         ## ------- Memory Decoder -------
 
-        'dim_compressed': 900,
 
+        'dim_compressed': 300,
 
         'com_dim_feedforward': 1024,
 
         'com_dim_feedforward': 1024,
 
         'com_num_heads': 8,
 
         'com_num_heads': 8,
 
         'com_dropout': 0.1,
 
         'com_dropout': 0.1,