File size: 1,252 Bytes
abd2a81 |
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 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 |
{
"data_dir_candidates": [
"/data/titane/user/nigirard/data",
"~/data",
"/data"
],
"num_workers": null, // If null, will use multiprocess.cpu_count() workers in total
"data_aug_params": {
"enable": true,
"vflip": true,
"affine": true,
"scaling": [0.75, 1.5], // Range of scaling factor to apply during affine transform. Set to None to not apply.
"color_jitter": true,
"device": "cuda"
},
"device": "cuda", // Only has effects when mode is val or test. When mode is train, always use CUDA
"use_amp": false, // Automatic Mixed Precision switch
"compute_seg": true,
"compute_crossfield": true,
"seg_params": {
"compute_interior": true,
"compute_edge": true,
"compute_vertex": false
},
"loss_params": {
"defaults_filepath": "configs/loss_params.json" // Path from the project's root to a JSON with default values for dataset_params
},
"optim_params": {
"defaults_filepath": "configs/optim_params.json" // Path from the project's root to a JSON with default values for optim_params
},
"polygonize_params": {
"defaults_filepath": "configs/polygonize_params.json" // Path from the project's root to a JSON with default values for polygonize_params
}
}
|