зеркало из https://github.com/microsoft/torchgeo.git
20 строки
349 B
YAML
20 строки
349 B
YAML
model:
|
|
class_path: SimCLRTask
|
|
init_args:
|
|
model: 'resnet18'
|
|
in_channels: 7
|
|
version: 1
|
|
layers: 2
|
|
hidden_dim: 8
|
|
output_dim: 8
|
|
weight_decay: 1e-6
|
|
memory_bank_size: 0
|
|
data:
|
|
class_path: SSL4EOLDataModule
|
|
init_args:
|
|
batch_size: 2
|
|
dict_kwargs:
|
|
root: 'tests/data/ssl4eo/l'
|
|
split: 'oli_sr'
|
|
seasons: 1
|