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