зеркало из https://github.com/microsoft/torchgeo.git
15 строки
284 B
YAML
15 строки
284 B
YAML
|
model:
|
||
|
|
class_path: PixelwiseRegressionTask
|
||
|
|
init_args:
|
||
|
|
model: 'unet'
|
||
|
|
backbone: 'resnet18'
|
||
|
|
in_channels: 3
|
||
|
|
loss: 'mae'
|
||
|
|
data:
|
||
|
|
class_path: InriaAerialImageLabelingDataModule
|
||
|
|
init_args:
|
||
|
|
batch_size: 1
|
||
|
|
patch_size: 2
|
||
|
|
dict_kwargs:
|
||
|
|
root: 'tests/data/inria'
|