зеркало из https://github.com/microsoft/torchgeo.git
19 строки
353 B
YAML
19 строки
353 B
YAML
|
program: # These are the arguments that define how the train.py script works
|
||
|
|
seed: 1337
|
||
|
|
overwrite: True
|
||
|
|
|
||
|
|
experiment:
|
||
|
|
task: cowc_counting
|
||
|
|
name: cowc_counting_test
|
||
|
|
module:
|
||
|
|
model: resnet18
|
||
|
|
learning_rate: 1e-3
|
||
|
|
learning_rate_schedule_patience: 2
|
||
|
|
datamodule:
|
||
|
|
batch_size: 32
|
||
|
|
num_workers: 4
|
||
|
|
|
||
|
|
trainer:
|
||
|
|
min_epochs: 15
|
||
|
|
gpus: 1
|