Kaynağa Gözat

modify rtdetr_r18 config

yjh0410 1 yıl önce
ebeveyn
işleme
b7e10afb6f
1 değiştirilmiş dosya ile 1 ekleme ve 1 silme
  1. 1 1
      config/model_config/rtdetr_config.py

+ 1 - 1
config/model_config/rtdetr_config.py

@@ -27,7 +27,7 @@ rtdetr_cfg = {
         'en_num_heads': 8,
         'en_num_layers': 1,
         'en_mlp_ratio': 4.0,
-        'en_dropout': 0.1,
+        'en_dropout': 0.0,
         'pe_temperature': 10000.,
         'en_act': 'gelu',
         # Transformer Decoder