| experiment_name: f5b_hyper_unetsempos1all_all_15ep_JustMag1c_R1 | |
| seed: None | |
| resume_from_checkpoint: false | |
| wandb: | |
| wandb_project: starcop-aviris-seg-vitek | |
| wandb_entity: dtacs | |
| images_logging: wandb | |
| dataloader: | |
| batch_size: 32 | |
| num_workers: 4 | |
| products_plot: | |
| - mag1c | |
| - label | |
| - pred | |
| - differences | |
| plot_samples: 8 | |
| dataset: | |
| input_products: | |
| - mag1c | |
| output_products: | |
| - labelbinary | |
| use_weight_loss: true | |
| weight_loss: weight_mag1c | |
| training_size: | |
| - 128 | |
| - 128 | |
| training_size_overlap: | |
| - 64 | |
| - 64 | |
| weight_sampling: true | |
| root_folder: /home/previtus/Permian/dataset | |
| train_csv: train.csv | |
| model: | |
| train: true | |
| test: true | |
| model_mode: segmentation_output | |
| model_type: unet_semseg | |
| semseg_backbone: mobilenet_v2 | |
| num_classes: 1 | |
| optimizer: adam | |
| lr: 0.0001 | |
| lr_decay: 0.5 | |
| lr_patience: 4 | |
| loss: BCEWithLogitsLoss | |
| pos_weight: 1 | |
| early_stopping_patience: 8 | |
| training: | |
| accelerator: gpu | |
| devices: 1 | |
| max_epochs: 15 | |
| val_check_interval: 0.5 | |
| train_log_every_n_steps: 10 | |