train_WSR.json

{
    "mode": "sr",
    "gpu_ids": [0],

    "scale": 4,
    "is_train": true,
    "use_chop": true,
    "rgb_range": 255,
    "self_ensemble": false,
    "save_image": true,

    "datasets": {
        "train": {
            "mode": "LRHR",
            "dataroot_HR": "/public/zhr/DIV2K/Augment/DIV2K_train_HR_aug/x4",
            "dataroot_LR": "/public/zhr/DIV2K/Augment/DIV2K_train_LR_aug/x4",
            "data_type": "npy",
            "n_workers": 16,
            "batch_size": 16,
            "LR_size": 40,
            "use_flip": true,
            "use_rot": true,
            "noise": "."
        },
        "val": {
            "mode": "LRHR",
            "dataroot_HR": "./results/HR/Set5/x4",
            "dataroot_LR": "./results/LR/LRBI/Set5/x4",
            "data_type": "img"
        }
    },

    "networks": {
        "which_model": "WSR",
        "num_features": 64,
        "in_channels": 3,
        "out_channels": 3,
        "num_groups": 10
    },

    "solver": {
        "type": "ADAM",
        "learning_rate": 0.0001,
        "weight_decay": 0,
        "lr_scheme": "MultiStepLR",
        "lr_steps": [2000],
        "lr_gamma": 0.5,
        "loss_type": "l1",
        "manual_seed": 0,
        "num_epochs": 2000,
        "skip_threshold": 3,
        "split_batch": 1,
        "save_ckp_step": 50,
        "save_vis_step": 1,
        "pretrain": null,
        "pretrained_path": "./experiments/WSR_in3f64_x4/epochs/last_ckp.pth",
        "wl_weights": [1.0, 1.0, 1.0]
    }
}