Przeglądaj źródła

set ffn_dim 2048 for RT-DETR-R50

yjh0410 1 rok temu
rodzic
commit
65fb49cfe8
1 zmienionych plików z 2 dodań i 2 usunięć
  1. 2 2
      config/model_config/rtdetr_config.py

+ 2 - 2
config/model_config/rtdetr_config.py

@@ -78,7 +78,7 @@ rtdetr_cfg = {
         'hidden_dim': 256,
         'en_num_heads': 8,
         'en_num_layers': 1,
-        'en_ffn_dim': 1024,
+        'en_ffn_dim': 2048,
         'en_dropout': 0.0,
         'pe_temperature': 10000.,
         'en_act': 'gelu',
@@ -86,7 +86,7 @@ rtdetr_cfg = {
         'transformer': 'rtdetr_transformer',
         'de_num_heads': 8,
         'de_num_layers': 6,
-        'de_ffn_dim': 1024,
+        'de_ffn_dim': 2048,
         'de_dropout': 0.0,
         'de_act': 'relu',
         'de_num_points': 4,