RTLucassen's picture
Upload 210 files
dc08fea verified
{"experiment_name": "0028_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": 3, "iterations": 500000, "iterations_per_update": 20, "iterations_per_checkpoint": 5000, "save_all_checkpoints": false, "label_type": "radians", "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": "CS", "model": {"patch_shape": 16, "input_dim": 3, "embed_dim": 256, "n_classes": 1, "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}