зеркало из https://github.com/microsoft/torchgeo.git
27 строки
508 B
YAML
27 строки
508 B
YAML
trainer:
|
|
gpus: 1
|
|
min_epochs: 20
|
|
max_epochs: 500
|
|
benchmark: True
|
|
|
|
experiment:
|
|
task: "oscd"
|
|
module:
|
|
loss: "jaccard"
|
|
segmentation_model: "unet"
|
|
encoder_name: "resnet18"
|
|
encoder_weights: null
|
|
learning_rate: 1e-3
|
|
learning_rate_schedule_patience: 6
|
|
verbose: false
|
|
in_channels: 26
|
|
num_classes: 2
|
|
num_filters: 128
|
|
ignore_zeros: True
|
|
datamodule:
|
|
train_batch_size: 2
|
|
num_workers: 6
|
|
val_split_pct: 0.1
|
|
bands: "all"
|
|
num_patches_per_tile: 128
|