зеркало из https://github.com/microsoft/torchgeo.git
18 строки
326 B
YAML
18 строки
326 B
YAML
trainer:
|
|
gpus: 1 # single GPU training
|
|
min_epochs: 10
|
|
max_epochs: 40
|
|
benchmark: True
|
|
|
|
experiment:
|
|
task: "resisc45"
|
|
module:
|
|
loss: "ce"
|
|
classification_model: "resnet18"
|
|
learning_rate: 1e-3
|
|
learning_rate_schedule_patience: 6
|
|
weights: "random"
|
|
datamodule:
|
|
batch_size: 128
|
|
num_workers: 6
|