| { | |
| "cfg": { | |
| "arch": "MaGGIe", | |
| "aspp": { | |
| "in_channels": 512, | |
| "out_channels": 512 | |
| }, | |
| "decoder": "res_shortcut_inst_matt_spconv_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, | |
| "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": 0.0, | |
| "loss_reweight_os8": true, | |
| "shm": { | |
| "dilation_kernel": 15, | |
| "lr_scale": 0.5, | |
| "max_n_pixel": 4000000, | |
| "mgm_weights": "" | |
| }, | |
| "sync_bn": true, | |
| "warmup_iters": 3000, | |
| "weights": "checkpoints/image/best_model.pth" | |
| } | |
| } |