|
|
@@ -78,7 +78,7 @@ rtdetr_cfg = {
|
|
|
'hidden_dim': 256,
|
|
|
'en_num_heads': 8,
|
|
|
'en_num_layers': 1,
|
|
|
- 'en_ffn_dim': 1024,
|
|
|
+ 'en_ffn_dim': 2048,
|
|
|
'en_dropout': 0.0,
|
|
|
'pe_temperature': 10000.,
|
|
|
'en_act': 'gelu',
|
|
|
@@ -86,7 +86,7 @@ rtdetr_cfg = {
|
|
|
'transformer': 'rtdetr_transformer',
|
|
|
'de_num_heads': 8,
|
|
|
'de_num_layers': 6,
|
|
|
- 'de_ffn_dim': 1024,
|
|
|
+ 'de_ffn_dim': 2048,
|
|
|
'de_dropout': 0.0,
|
|
|
'de_act': 'relu',
|
|
|
'de_num_points': 4,
|