# config.yaml

net:
  path: "geoseg/models/UNetFormer.py"  # 替换为你的模型文件路径
  class: "UNetFormer"  # 模型类名
loss:
  path: "geoseg/losses/useful_loss.py"  # 替换为你的损失函数文件路径
  class: "EdgeLoss"  # 损失函数类名
num_classes: 2
classes: ['background', 'pv']
optimizer: torch.optim.Adam
lr_scheduler: torch.optim.lr_scheduler.StepLR
train_loader:
  batch_size: 8  # 训练批处理大小
  shuffle: True  # 是否随机打乱数据
  num_workers: 4  # 数据加载的并行线程数

val_loader:
  batch_size: 8  # 验证批处理大小
  shuffle: False  # 验证时不需要打乱数据
  num_workers: 4  # 数据加载的并行线程数
save_top_k: 3
monitor: 'val_mIoU'
save_last: True
monitor_mode: 'max'
weights_path: 'pretrain_weights/stseg_small.pth'
weights_name: 'stseg_small'
log_name: 'loveda'
max_epoch: 200
gpus: [0]
check_val_every_n_epoch: 10
use_aux_loss: False
pretrained_ckpt_path: None
resume_ckpt_path: None