chuonghm's picture
Push model using huggingface_hub.
6337268 verified
{
"cfg": {
"arch": "MaGGIe_Temp",
"aspp": {
"in_channels": 512,
"out_channels": 512
},
"decoder": "res_shortcut_inst_matt_spconv_temp_22",
"decoder_args": {
"atten_block": 2,
"atten_dim": 128,
"atten_head": 1,
"atten_stride": 1,
"detail_mask_dropout": 0.1,
"final_channel": 64,
"freeze_detail_branch": false,
"head_channel": 120,
"max_inst": 10,
"temp_method": "bi_fusion",
"use_id_pe": true,
"warmup_detail_iter": 3000,
"warmup_mask_atten_iter": 0
},
"encoder": "res_shortcut_embed_29",
"encoder_args": {
"num_embed": 3,
"num_mask": 10,
"pretrained": true
},
"having_unused_params": false,
"loss_alpha_grad_w": 0.05,
"loss_alpha_lap_w": 0.05,
"loss_alpha_type": "l1",
"loss_alpha_w": 1.0,
"loss_atten_w": 5.0,
"loss_dtSSD_w": 1.0,
"loss_reweight_os8": true,
"shm": {
"dilation_kernel": 15,
"lr_scale": 0.5,
"max_n_pixel": 4000000,
"mgm_weights": ""
},
"sync_bn": true,
"warmup_iters": 500,
"weights": "checkpoints/video/best_model.pth"
}
}