|
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 |
|
|