소스 검색

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