Skip to content

Commit 57f928d

Browse files
committed
update
1 parent 77ed89e commit 57f928d

File tree

1 file changed

+0
-5
lines changed

1 file changed

+0
-5
lines changed

rtdetr_pytorch/configs/rtdetr/include/rtdetr_r50vd.yml

Lines changed: 0 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -49,7 +49,6 @@ RTDETRTransformer:
4949
hidden_dim: 256
5050
num_levels: 3
5151

52-
num_classes: 80
5352
num_queries: 300
5453

5554
num_decoder_layers: 6
@@ -63,13 +62,9 @@ use_focal_loss: True
6362

6463
RTDETRPostProcessor:
6564
num_top_queries: 300
66-
# num_classes: 80
67-
# use_focal_loss: True
68-
# remap_mscoco_category: True
6965

7066

7167
SetCriterion:
72-
# num_classes: 80
7368
weight_dict: {loss_vfl: 1, loss_bbox: 5, loss_giou: 2,}
7469
losses: ['vfl', 'boxes', ]
7570
alpha: 0.75

0 commit comments

Comments
 (0)