Daniele Picone
Updated to 0.10.0
d9b5f18
{
"backend": {
"id": "torch",
"device": "none"
},
"dataset_training": {
"dataset_list": [
{
"image": {
"id": null,
"dataset": "chikusei",
"dataset_id": 0
},
"preprocessing": {
"crop_row": [
64,
1664
],
"crop_col": [
64,
2240
],
"crop_chn": [
8,
72,
4
],
"crop_acq": null,
"select_chn": null,
"normalize_intensity": true,
"equalize_dynamic": true,
"dynamic_range": [
0.05,
0.95
]
}
}
],
"patch_size": 64,
"stride": 16,
"horizontal": true,
"vertical": true,
"rotation": true
},
"dataset_validation": {
"dataset_list": [
{
"image": {
"id": null,
"dataset": "chikusei",
"dataset_id": 0
},
"preprocessing": {
"crop_row": [
1664,
2048
],
"crop_col": [
64,
2240
],
"crop_chn": [
8,
72,
4
],
"crop_acq": null,
"select_chn": null,
"normalize_intensity": true,
"equalize_dynamic": true,
"dynamic_range": [
0.05,
0.95
]
}
}
],
"patch_size": 64,
"stride": 64,
"horizontal": false,
"vertical": false,
"rotation": false
},
"model": {
"linear_id": "deblurring",
"kernel_id": 0,
"color_filter_id": null,
"spectral_responses_id": null,
"spectral_responses_type": "multispectral",
"scale_ratio": null,
"lagrange_multipliers": []
},
"noise": {
"id": "gaussian",
"snr": 25.0,
"mean_scaled_snr": true
},
"processing": {
"data_functional": "squared_l2",
"priors": [
{
"id": "pnp",
"unfolded_stages": 4,
"denoiser_deep": {
"id": "unet",
"in_channels": 16
},
"regularization_parameter": 1.0,
"noise_level": 0.03
}
],
"solver": {
"id": "admm",
"penalization_parameter_list": [
0.1
],
"penalization_parameter": 0.1,
"penalization_parameter_dual": 0.1,
"relaxation_parameter": 0.1,
"sub_problem_solver": "auto",
"max_iterations": 1,
"verbose": true,
"g_first": false,
"early_stop": false,
"nan_stop": false
}
},
"dataloader_options": {
"shuffle_train": true,
"shuffle_validation": false,
"num_workers": 4,
"batch_size": 48
},
"trainable_parameters": {
"regularization_parameter": false,
"noise_level": false,
"relaxation_parameter": true,
"penalization_parameter": true,
"penalization_parameter_dual": false
},
"training_options": {
"optimizer": "adam",
"optimizer_momentum": 0.9,
"learning_rate": 0.01,
"loss_validation": "mse",
"loss_test": [
"mse",
"ssim",
"psnr"
],
"scheduler": {
"id": "reduce_on_plateau",
"max_iterations": 50,
"reduction_factor": 0.1,
"patience": 10,
"tolerance": 1e-6
}
},
"training_logger": {
"logger": "csv",
"maximum_time": "00:00:40:00",
"early_stopping_delta": 1e-6,
"early_stopping_patience": 20,
"tqdm_refresh_rate": 10
}
}