detr_custom / config.json
zeroone012012's picture
modified *.json
2a95a10
raw
history blame contribute delete
426 Bytes
{
"hidden_dim": 256,
"nheads": 8,
"num_encoder_layers": 6,
"num_decoder_layers": 6,
"dim_feedforward": 2048,
"dropout": 0.1,
"activation": "relu",
"num_classes": 10,
"num_queries": 100,
"aux_loss": true,
"bbox_loss_coef": 5,
"giou_loss_coef": 2,
"eos_coef": 0.1,
"set_cost_class": 0.1,
"set_cost_bbox": 5.0,
"set_cost_giou": 2.0,
"scale_loss_weight": 0.1
}