| { | |
| "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 | |
| } |