model_config {
  arch: "resnet",
  n_layers: 50
  use_bias: true
  use_batch_norm: true
  all_projections: true
  input_image_size: "3,100,100"
}
train_config {
  train_dataset_path: "/workspace/tlt-experiments/Classification/Dataset_8_5_2020/SolderBall/train"
  val_dataset_path: "/workspace/tlt-experiments/Classification/Dataset_8_5_2020/SolderBall/val"
  pretrained_model_path: "/workspace/tlt-experiments/Classification/Classifications_pretrained/tlt_resnet50_classification_v1/resnet50.hdf5"
  optimizer: "sgd"
  batch_size_per_gpu: 64
  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.007
    #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: "/workspace/tlt-experiments/Classification/Dataset_23_4_2020/resnet50/resnet_015.tlt"
  top_k: 3
  batch_size: 4
  n_workers: 4
}