File size: 1,971 Bytes
dc08fea |
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 42 43 44 45 46 47 48 49 50 |
{
"experiment_name": "0046_experiment",
"experiments_folder": "/hpc/dlab_skin_ai/projects/tissue_matching/models/rotation",
"norm_image_folder": "/hpc/dlab_skin_ai/projects/tissue_matching/data/normalized/images",
"norm_segmentation_folder": "/hpc/dlab_skin_ai/projects/tissue_matching/data/normalized/annotations",
"unnorm_image_folder": "/hpc/dlab_skin_ai/projects/tissue_matching/data/unnormalized/images",
"unnorm_segmentation_folder": "/hpc/dlab_skin_ai/projects/tissue_matching/data/unnormalized/annotations",
"labels_path": [
"/hpc/dlab_skin_ai/projects/tissue_matching/data/sheets/image_rotations_IHC.json"
],
"selection_path": "/hpc/dlab_skin_ai/projects/tissue_matching/data/sheets/assigned_split.json",
"mapping_path": "/hpc/dlab_skin_ai/projects/tissue_matching/data/sheets/patient_mapping.json",
"seed": 1,
"iterations": 500000,
"iterations_per_update": 20,
"iterations_per_checkpoint": 5000,
"save_all_checkpoints": false,
"label_type": "coordinate",
"gaussian_stdev": null,
"augmentation": {
"oversample_small_angles": true,
"small_angle_prob": 0.25,
"small_angle_range": 30,
"brightness": 0.2,
"contrast": 0.2,
"saturation": 0.2,
"hue": 0.05
},
"learning_rate": 2e-05,
"weight_decay": 0.01,
"lr_patience": 5,
"lr_factor": 0.5,
"loss_function": "CS2",
"model": {
"patch_shape": 16,
"input_dim": 3,
"embed_dim": 256,
"n_classes": 2,
"depth": 14,
"n_heads": 4,
"mlp_ratio": 5,
"pytorch_attn_imp": false,
"init_values": 1e-05,
"max_tiles": 15000,
"tile_dropout_prob": 0.0
},
"pretrained_weights_path": "/hpc/dlab_skin_ai/repositories/tissue_matching/rotation/vit_wee_patch16_reg1_gap_256.sbb_in1k.pth",
"num_workers": 4,
"pin_memory": false
} |