DeepSign/config.yaml at main · 23daVinci/DeepSign · GitHub

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
51
52
53
54
# config.yaml

project:
  name: DeepSign
  seed: 42
  device: "cpu"  # or "cuda"

data:
  dataset_name: IDCAR 2011
  train_dir: "../data/sign_data/train"
  test_dir: "../data/sign_data/test"
  train_pairs: "../data/sign_data/train_data.csv"
  test_pairs: "../data/sign_data/test_data.csv"
  train_serialized_path: "../data/train_pairs.tfrecord"
  img_height: 128
  img_width: 256
  batch_size: 32
  num_workers: 1
  shuffle_buffer_size: 2048
  TFRecord_buffer_size: 33554432  # 32 MB

model:
  architecture: "SiameseNet"
  embedding_dim: 128
  pretrained: False
  save_path: "../artifacts/trained_model.keras"

training:
  val_fraction: 0.1
  epochs: 5
  learning_rate: 0.0001
  loss: 'binary_crossentropy'
  optimizer: "adam"
  weight_decay: 1e-5
  early_stopping: True
  patience: 5
  lr_scheduler: "StepLR"
  step_size: 10
  gamma: 0.1

logging:
  log_dir: "logs/"
  use_wandb: True
  wandb_project: "signature-verification"

checkpoint:
  save_dir: "checkpoints/"
  save_best_only: True
  monitor: "val_loss"

augmentation:
  horizontal_flip: True
  rotation_range: 15
  brightness_range: [0.8, 1.2]