models/hparams.json

{
    "num_epochs": 10,
    "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": "./models",
    "save_dir": "./epoch_outputs",
    "num_saved": 30,
    "good_ex": [
        20,
        267,
        511
    ],
    "image_size": 1024,
    "hidden_dim": 64,
    "use_mask_aux_loss": true,
    "use_mask_final_loss": true,
    "use_box_loss": false,
    "lr": 0.0001,
    "weight_decay": 1e-05
}