Computer_Vision_for_Industrial_Inspection/2/resnet_config.txt


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50

model_config {
  arch: "resnet"
  n_layers: 50
  use_batch_norm: False
  freeze_blocks: 0
  input_image_size: "3,224,224"
}

train_config {
  train_dataset_path: "/workspace/tao-experiments/data/train"
  val_dataset_path: "/workspace/tao-experiments/data/val"
  pretrained_model_path: "/workspace/tao-experiments/classification/pretrained_resnet50/pretrained_classification_vresnet50/resnet_50.hdf5"
  optimizer {
    sgd {
    lr: 0.01
    decay: 0.0
    momentum: 0.9
    nesterov: False
    }
  }
  n_epochs: 15
  batch_size_per_gpu: 32
  n_workers: 8
  enable_random_crop: False
  enable_center_crop: False
  enable_color_augmentation: False
  preprocess_mode: "caffe"
  reg_config {
    type: "L2"
    scope: "Conv2D, Dense"
    weight_decay: 0.00005
  }
  lr_config {
    step {
      learning_rate: 0.006
      step_size: 10
      gamma: 0.1
    }
  }
}

eval_config {
  eval_dataset_path: "/workspace/tao-experiments/data/val"
  model_path: "/workspace/tao-experiments/classification/resnet50/weights/resnet_005.hdf5"
  top_k: 1
  batch_size: 32
  n_workers: 8
  enable_center_crop: False
}