File size: 735 Bytes
3e1d5f8 | 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 | {
"data_path": "/home/satoshi.tsutsui/projects/wbcas/dataset_txt/pbc_attr_v1_ccrop_all.csv",
"data_root": "/home/satoshi.tsutsui/satoshissd/PBC/pbcseg_final_v1/",
"model_name": "facebook/mask2former-swin-tiny-ade-semantic",
"resolution": 1024,
"out_resolution": 360,
"num_classes": 6,
"ignore_index": 0,
"no_flip": false,
"use_crop": false,
"use_color": true,
"freeze_encoder": false,
"freeze_decoder": false,
"lr": 3e-05,
"weight_decay": 0.01,
"epochs": 50,
"batch_size": 16,
"num_workers": 1,
"pflip": 0.5,
"grad_clip": 1.0,
"seed": 42,
"label_smoothing": 0.1,
"device": "cuda",
"save_dir": "./experiments",
"exp_name": "m2f_tiny_1024"
} |