# Sample model config for to instantiate a resnet18 model freeze blocks 0, 1
# with all shortcuts having projection layers.

model_config {
  arch: "vanilla_unet_dynamic"
  use_batch_norm: true
  initializer: HE_UNIFORM
  training_precision {
    backend_floatx: FLOAT32
  }
  load_graph: true
  model_input_height: 544
  model_input_width: 960
  model_input_channels: 3
}


training_config {
  batch_size: 4
  epochs: 200
  visualizer{
    enabled: true
  }
  log_summary_steps: 10
  checkpoint_interval: 5
  loss: "cross_entropy"
  learning_rate:0.0001
  regularizer {
    type: L2
    weight: 0
  }
  optimizer {
    adam {
      epsilon: 9.99999993923e-09
      beta1: 0.899999976158
      beta2: 0.999000012875
    }
  }
}

dataset_config {
    dataset: "custom"
    augment: True
    resize_padding: True
    resize_method: "BILINEAR"
    augmentation_config {
      spatial_augmentation {
        hflip_probability : 0.5
        vflip_probability : 0.5
        crop_and_resize_prob : 0.5
      }
      brightness_augmentation {
        delta: 0.25
      }
    }

    input_image_type: "color"
    train_images_path:"/workspace/tao-experiments/data/real_world_dataset/train_image_png_resize"
    train_masks_path:"/workspace/tao-experiments/data/real_world_dataset/train_labels"
    val_images_path:"/workspace/tao-experiments/data/real_world_dataset/val_image_png_resize"
    val_masks_path:"/workspace/tao-experiments/data/real_world_dataset/val_labels"
    test_images_path:"/workspace/tao-experiments/data/real_world_dataset/val_image_png_resize"
    data_class_config {
      target_classes {
      name: "free_space"
      mapping_class: "free_space"
      label_id: 0
    }
    target_classes {
      name: "blocked"
      mapping_class: "blocked"
      label_id: 1
    }
  }
}