model_config{
  arch: "resnet"
  n_layers: 50 
  use_bias: true 
  use_batch_norm: true 
  all_projections: true
  input_image_size: "3,200,200"
}
train_config{
  train_dataset_path: "/workspace/tlt-experiments/Classification/Dataset_23_4_2020/train"
  val_dataset_path: "/workspace/tlt-experiments/Classification/Dataset_23_4_2020/val"
  pretrained_model_path: "/home/itc/NVIDIA-tlt/tlt-experiments/Classification/Dataset_23_4_2020/resnet50_200px/weights/resnet_025_prune/weights/resnet_015.tlt"
  optimizer: "sgd"
  batch_size_per_gpu: 32
  n_epochs: 25
  n_workers: 4

  # regularizer
  reg_config {
    type: "L2"
    scope: "Conv2D,Dense"
    weight_decay: 0.00005
  }

  # learning_rate
  lr_config {
    scheduler: "step"
    learning_rate: 0.006
    #soft_start: 0.056
    #annealing_points: "0.3, 0.6, 0.8"
    #annealing_divider: 10
    step_size: 10
    gamma: 0.1
  }
}
eval_config{
  eval_dataset_path: "/workspace/tlt-experiments/Classification/Dataset_23_4_2020/test"
  model_path: "/home/itc/NVIDIA-tlt/tlt-experiments/Classification/Dataset_23_4_2020/resnet50_200px/weights/resnet_025_prune/weights/resnet_015.tlt"
  top_k: 3
  batch_size: 4
  n_workers: 4
}