text: _base_ = [
      '../_base_/models/ocrnet_hr18.py', '../_base_/datasets/cityscapes_rain.py',
      '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py'
  ]
  norm_cfg = dict(type='SyncBN', requires_grad=True)
  model = dict(
      pretrained='open-mmlab://msra/hrnetv2_w48',
      backbone=dict(
          extra=dict(
              stage2=dict(num_channels=(48, 96)),
              stage3=dict(num_channels=(48, 96, 192)),
              stage4=dict(num_channels=(48, 96, 192, 384)))),
      decode_head=[
          dict(
              type='FCNHead',
              in_channels=[48, 96, 192, 384],
              channels=sum([48, 96, 192, 384]),
              input_transform='resize_concat',
              in_index=(0, 1, 2, 3),
              kernel_size=1,
              num_convs=1,
              norm_cfg=norm_cfg,
              concat_input=False,
              dropout_ratio=-1,
              num_classes=19,
              align_corners=False,
              loss_decode=dict(
                  type='CrossEntropyLoss', use_sigmoid=False, loss_weight=0.4)),
          dict(
              type='OCRHead',
              in_channels=[48, 96, 192, 384],
              channels=512,
              ocr_channels=256,
              input_transform='resize_concat',
              in_index=(0, 1, 2, 3),
              norm_cfg=norm_cfg,
              dropout_ratio=-1,
              num_classes=19,
              align_corners=False,
              loss_decode=dict(
                  type='CrossEntropyLoss', use_sigmoid=False, loss_weight=1.0))
      ])
  resume_from = 'work_dirs/ocrnet_hr48_512x1024_100k_cityscapes_blurred_ft/iter_100000.pth'
  
  optimizer_config = dict(type='Fp16OptimizerHook', loss_scale=512.)
  runner = dict(type='IterBasedRunner', max_iters=120000)
  data = dict(
      samples_per_gpu=4)
  checkpoint_config = dict(  # Config to set the checkpoint hook, Refer to https://github.com/open-mmlab/mmcv/blob/master/mmcv/runner/hooks/checkpoint.py for implementation.
      by_epoch=False,  # Whethe count by epoch or not.
      interval=2000)  # The save interval.