فهرست منبع

modify rtdetr_r18 config

yjh0410 1 سال پیش
والد
کامیت
b7e10afb6f
1فایلهای تغییر یافته به همراه1 افزوده شده و 1 حذف شده
  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