| experiment_name = "unet_dec_21_lsdb_only" | |
| base_dir = "/exports/csce/eddie/eng/groups/DunnGroup/matthew/models_gelgenie" | |
| [processing] | |
| base_hardware = "EDDIE" | |
| device = "GPU" | |
| pe = 1 | |
| memory = 64 | |
| [data] | |
| n_channels = 1 | |
| batch_size = 2 | |
| num_workers = 1 | |
| val_percent = 10 | |
| dir_train_mask = [ "/exports/csce/eddie/eng/groups/DunnGroup/matthew/gel_data/lsdb_gels/masks",] | |
| dir_train_img = [ "/exports/csce/eddie/eng/groups/DunnGroup/matthew/gel_data/lsdb_gels/images",] | |
| dir_val_img = [ "/exports/csce/eddie/eng/groups/DunnGroup/matthew/gel_data/lsdb_gels/val_images",] | |
| dir_val_mask = [ "/exports/csce/eddie/eng/groups/DunnGroup/matthew/gel_data/lsdb_gels/val_masks",] | |
| split_training_dataset = false | |
| apply_augmentations = true | |
| padding = true | |
| individual_padding = false | |
| weak_augmentations = false | |
| minmax_norm = false | |
| [model] | |
| model_name = "smp_unet" | |
| classes = 2 | |
| in_channels = 1 | |
| encoder_name = "resnet18" | |
| [training] | |
| loss = [ "dice", "crossentropy",] | |
| loss_component_weighting = [ 1, 1,] | |
| class_loss_weighting = false | |
| class_loss_weight_damper = [ 1.0, 1.0,] | |
| lr = 0.0001 | |
| epochs = 600 | |
| grad_scaler = true | |
| load_checkpoint = false | |
| optimizer_type = "adam" | |
| scheduler_type = "CosineAnnealingWarmRestarts" | |
| save_checkpoint = true | |
| checkpoint_frequency = 1 | |
| wandb_track = true | |
| model_cleanup_frequency = 20 | |
| wandb_id = "3etx3f9s" | |
| [training.scheduler_specs] | |
| restart_period = 100 | |