зеркало из https://github.com/microsoft/torchgeo.git
21 строка
382 B
YAML
21 строка
382 B
YAML
model:
|
|
class_path: MoCoTask
|
|
init_args:
|
|
model: 'resnet18'
|
|
in_channels: 4
|
|
data:
|
|
class_path: ChesapeakeCVPRDataModule
|
|
init_args:
|
|
train_splits:
|
|
- 'de-test'
|
|
val_splits:
|
|
- 'de-test'
|
|
test_splits:
|
|
- 'de-test'
|
|
batch_size: 2
|
|
patch_size: 64
|
|
class_set: 5
|
|
use_prior_labels: True
|
|
dict_kwargs:
|
|
root: 'tests/data/chesapeake/cvpr'
|