architecture: "Unet" | |
model_params: | |
encoder_name: "resnet50" | |
encoder_weights: "swsl" | |
in_channels: 11 | |
classes: 11 | |
device: "cuda" | |
seed: 0 | |
save_path: "./checkpoints/" | |
training: | |
dataloader_params: | |
landsat: | |
root: "./Megacity2035/train" | |
filename_glob: "**/landsat8_2021.tif" | |
exclude: null | |
esa: | |
root: "./Megacity2035/train" | |
filename_glob: "**/ESAv200_.tif" | |
exclude: null | |
sensor_type: "OLITIRS" | |
tile_per_loader: 24 | |
length: 50000 | |
batch_size: 128 | |
num_workers: 8 | |
trainer_params: | |
epoch: 100 | |
gradient_accumulation_factor: null | |
loss_fn_params: | |
weights: [1.0, 1.0] | |
FocalLoss: | |
mode: "multiclass" | |
alpha: null | |
gamma: 2.0 | |
normalized: False | |
TverskyLoss: | |
mode: "multiclass" | |
alpha: 0.3 | |
beta: 0.7 | |
gamma: 2.0 | |
optimizer_params: | |
lr: 5.0e-4 | |
weight_decay: 1.0e-5 | |
scheduler_params: | |
step_size: 30 | |
gamma: 0.5 | |
last_epoch: -1 | |
validate_params: | |
root: "./Megacity2035/val" | |
img_glob: "**/landsat8_2021.tif" | |
gt_glob: "**/ESAv200_.tif" | |
logger_params: | |
PROJECT: "GUC_master" | |
NAME: "dummy" |