| { | |
| "paths": { | |
| "data_path": "", | |
| "save_path": "" | |
| }, | |
| "subject_list": ["rca", "ladprox", "laddist", "lcx", "normal"], | |
| "CV": { | |
| "val_subject_id": 0 | |
| }, | |
| "network_prm": { | |
| "input_dim": [128, 128, 1], | |
| "n_levels": 4, | |
| "n_init_filters": 16, | |
| "in_skip": true, | |
| "attention": true, | |
| "kernel_size": 3, | |
| "conv_stride": 1, | |
| "upsampling_stride": [2, 2], | |
| "pool_size": [2, 2], | |
| "pool_stride": 1, | |
| "padding": "same", | |
| "act": "linear" | |
| }, | |
| "learning_prm": { | |
| "batch_size": 50, | |
| "lr": 1e-4, | |
| "drp": 0.05, | |
| "loss": "mean_squared_error", | |
| "metrics": ["mae"], | |
| "n_epochs": 10 | |
| }, | |
| "tr_phase": true, | |
| "model_name": "CF2D_L2Loss", | |
| "weight_name": "" | |
| } |