зеркало из https://github.com/microsoft/torchgeo.git
21 строка
387 B
YAML
21 строка
387 B
YAML
|
module:
|
||
|
|
_target_: torchgeo.trainers.MoCoTask
|
||
|
|
model: "resnet18"
|
||
|
|
in_channels: 4
|
||
|
|
|
||
|
|
datamodule:
|
||
|
|
_target_: torchgeo.datamodules.ChesapeakeCVPRDataModule
|
||
|
|
root: "tests/data/chesapeake/cvpr"
|
||
|
|
download: false
|
||
|
|
train_splits:
|
||
|
|
- "de-test"
|
||
|
|
val_splits:
|
||
|
|
- "de-test"
|
||
|
|
test_splits:
|
||
|
|
- "de-test"
|
||
|
|
batch_size: 2
|
||
|
|
patch_size: 64
|
||
|
|
num_workers: 0
|
||
|
|
class_set: 5
|
||
|
|
use_prior_labels: True
|