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