dataset_config {
  augment: true
  dataset: "custom"
  input_image_type: "color"
  train_images_path: "/bdd100k/images/10k/train"
  train_masks_path: "/bdd100k/labels/sem_seg/masks/train_255to19"
  val_images_path: "/bdd100k/images/10k/val"
  val_masks_path: "/bdd100k/labels/sem_seg/masks/val_255to19"
  test_images_path: "/bdd100k/images/10k/val"
  data_class_config {
    target_classes {
      name: "road"
      mapping_class: "road"
    }
    target_classes {
      name: "sidewalk"
      label_id: 1
      mapping_class: "sidewalk"
    }
    target_classes {
      name: "building"
      label_id: 2
      mapping_class: "building"
    }
    target_classes {
      name: "wall"
      label_id: 3
      mapping_class: "wall"
    }
    target_classes {
      name: "fence"
      label_id: 4
      mapping_class: "fence"
    }
    target_classes {
      name: "pole"
      label_id: 5
      mapping_class: "pole"
    }
    target_classes {
      name: "traffic light"
      label_id: 6
      mapping_class: "traffic light"
    }
    target_classes {
      name: "traffic sign"
      label_id: 7
      mapping_class: "traffic sign"
    }
    target_classes {
      name: "vegetation"
      label_id: 8
      mapping_class: "vegetation"
    }
    target_classes {
      name: "terrain"
      label_id: 9
      mapping_class: "terrain"
    }
    target_classes {
      name: "sky"
      label_id: 10
      mapping_class: "sky"
    }
    target_classes {
      name: "person"
      label_id: 11
      mapping_class: "person"
    }
    target_classes {
      name: "rider"
      label_id: 12
      mapping_class: "rider"
    }
    target_classes {
      name: "car"
      label_id: 13
      mapping_class: "car"
    }
    target_classes {
      name: "truck"
      label_id: 14
      mapping_class: "truck"
    }
    target_classes {
      name: "bus"
      label_id: 15
      mapping_class: "bus"
    }
    target_classes {
      name: "train"
      label_id: 16
      mapping_class: "train"
    }
    target_classes {
      name: "motorcycle"
      label_id: 17
      mapping_class: "motorcycle"
    }
    target_classes {
      name: "bicycle"
      label_id: 18
      mapping_class: "bicycle"
    }
    target_classes {
      name: "unknown"
      label_id: 19
      mapping_class: "unknown"
    }
  }
}
model_config {
  num_layers: 18
  use_batch_norm: true
  training_precision {
    backend_floatx: FLOAT32
  }
  arch: "resnet"
  model_input_height: 512
  model_input_width: 512
  model_input_channels: 3
}
training_config {
  batch_size: 4
  regularizer {
    type: L2
  }
  optimizer {
    adam {
      epsilon: 9.99999993922529e-09
      beta1: 0.8999999761581421
      beta2: 0.9990000128746033
    }
  }
  checkpoint_interval: 5
  log_summary_steps: 175
  learning_rate: 0.0010000000474974513
  loss: "cross_entropy"
  epochs: 240
}