train_config.json

{
    "seed": 0,
    "saved_dir" : "./results",
    "data_root" : "../input/data",
    "train_json_path" : "./splited_json/train0_test.json",
    "val_json_path" : "./splited_json/valid_split_0.json",
    "train_dataset" : {"name" : "TrainDataset",
                       "args" : {"mode" : "train",
                                 "cutmix_prob" : 0.4,
                                 "mixup_prob" : 0}
                      },
    "val_dataset" : {"name" : "TrainDataset",
                     "args" : {"mode" : "validation",
                               "cutmix_prob" : 0,
                               "mixup_prob" : 0}
                    },
    "train_dataloader": {"args" : {"batch_size" : 4,
                                   "num_workers" : 4,
                                   "shuffle" : true}},
    "val_dataloader" : {"args" : {"batch_size" : 4,
                                  "num_workers" : 4,
                                  "shuffle" : false}},
    "test_dataset" : "TestDataset",
    "train_augmentation": {"name" : "JYAugmentation",
                           "args" : {"data_root" : "../input/data",
                                     "json_dir" : "./splited_json/train_split_0.json"}},
    "val_augmentation": {"name" : "BaseAugmentation",
                         "args" : {}},
    "num_epochs": 400,
    "model": {"name" : "HRNetOCR",
              "args" : {"num_classes" : 11,
                        "target_size" : 512}},
    "resume_from": false,
    "resume_mode": "all",
    "checkpoint_path": "",

    "criterion": {"name" : "DiceCELoss",
                  "args": {
                      "dice_weight": 0.2,
                      "focal_weight": 0.8 
                  }},
    "optimizer": {"name" : "SGD",
                  "args" : {"lr" : 0.001}},
    "scheduler":{"name":"ReduceLROnPlateau",
        "args":{"mode" : "max",
                "factor" : 0.9,
                "patience" : 3}},
    "val_every" : 1,
    "wandb_prj_name": "jy_seg_dpt",
    "wandb_run_name": "hrnet_ocr",
    "save_mode": "miou",
    "num_to_remain": 3,
    "fp16": true
}