35 lines
977 B
YAML
35 lines
977 B
YAML
|
# 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
|