diff --git a/dygraph/configs/ocrnet/ocrnet_hrnetw18_cityscapes_1024x512_40k.yml b/dygraph/configs/ocrnet/ocrnet_hrnetw18_cityscapes_1024x512_40k.yml index e71271c1e59930cfbb0bcd68aab31b3706a82374..a8676a47d5341f1522fca1b8e812b620bf986023 100644 --- a/dygraph/configs/ocrnet/ocrnet_hrnetw18_cityscapes_1024x512_40k.yml +++ b/dygraph/configs/ocrnet/ocrnet_hrnetw18_cityscapes_1024x512_40k.yml @@ -1,44 +1,13 @@ +_base_: '../_base_/cityscapes.yml' + batch_size: 2 iters: 40000 -train_dataset: - type: Cityscapes - dataset_root: data/cityscapes - transforms: - - type: ResizeStepScaling - min_scale_factor: 0.5 - max_scale_factor: 2.0 - scale_step_size: 0.25 - - type: RandomPaddingCrop - crop_size: [1024, 512] - - type: RandomHorizontalFlip - - type: Normalize - mode: train - -val_dataset: - type: Cityscapes - dataset_root: data/cityscapes - transforms: - - type: Normalize - mode: val - model: type: OCRNet backbone: type: HRNet_W18 - backbone_pretrianed: None num_classes: 19 - in_channels: 270 + backbone_channels: [270] + backbone_pretrained: pretrained_model/hrnet_w18_imagenet model_pretrained: None - -optimizer: - type: sgd - -learning_rate: - value: 0.01 - decay: - type: poly - power: 0.9 - -loss: - type: CrossEntropy