task: Type
modelname: MobilenetV3
output: output
checkpoint: "output/ErrorType/85804_model=MobilenetV3-ep=3000-block=4/model_best.pth.tar"
gpu: [1]
data:
  train: ../data/Fifth_data/ErrorType
  val: ../data/Fifth_data/ErrorType
  test: ../data/Fifth_data/ErrorType
train:
  epochs: 3000
  start-epoch: 0
  batch-size: 256
  worker: 16
  resume: ''
  augment: True
  size: 64
  confidence: False
  weight: [1., 1., 1., 1., 1., 1., 1.]  #Crack, Double, Empty, Flip, Leave, Scratch
predict:
  batch-size: 256
  worker: 16
  cam: False
  normalize: True
  save: False
optimizer:
  type: 'SGD'
  lr: 0.1
  momentum: 0.9
  weight_decay: 0.0001
loss:
  gamma: 2.
  alpha: 0.8
model:
  blocks: 4
  class: 7
etc:
  tensorboard: False
  print_freq: 10