config.json

{
    "Taining_Dir": "data/train_val",
    "Training_Tensor_Dir": "data/Training_Tensor_data",
    "Testing_Tensor_Dir":   "data/Testing_Tensor_data",
    "Label_Path": "data/train_val.csv",
    "Test_Dir": "data/test",
    "_comment_": "The shape of the tensor feed into the network (namly HxW)",
    "H": 256,
    "W": 256,
    "_comment": "The GPU param is to the which GPU U want to use. If this is only one gpu, set it to 0.",
    "GPU": "0",
    "__comment": "The DEVICE param is set to indicate whether U want to use the GPU or CPU. If U wang to use GPU, then 'cuda' else 'cpu'. ",
    "DEVICE": "cuda",
    "num_workers": 0,
    "epoch": 200,
    "lr": 1e-3,
    "Weight_Decay": 0.01,
    "batch_size": 16,
    "gray": false ,
    "Commment:": "LZX pls use: C:\\Users\\ZXLi\\Anaconda3\\envs\\pytorch\\python",
    "Commment_:": "WMK pls use: /home/wangmingke/anaconda3/envs/pytorch/bin/python",
    "python_path": "/home/wangmingke/anaconda3/envs/pytorch/bin/python"
}