File size: 627 Bytes
6d07676 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 | {
"num_epochs": 5,
"batch_size": 4,
"use_amp": false,
"use_wandb": false,
"use_boxes": false,
"use_masks": true,
"patience": 3,
"task_name": "mask_sam_auto_prompt_ViT-B",
"model_save_path": "./outputs/models",
"save_dir": "./outputs/epoch_outputs",
"num_saved": 7,
"good_ex": [
20,
267,
511
],
"image_size": 1024,
"hidden_dim": 64,
"low_res_metric": true,
"useful": false,
"augmentation": null,
"use_mask_aux_loss": true,
"use_mask_final_loss": true,
"use_box_loss": false,
"lr": 0.0001,
"weight_decay": 1e-05
} |