Преглед на файлове

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