config.yaml 558 B

1234567891011121314151617181920212223242526272829
  1. ## Data loader
  2. batch_size: 200
  3. ## Training details
  4. max_epoch: 15
  5. trainfunc: softmax
  6. ## Optimizer
  7. scheduler: steplr
  8. optimizer: adam
  9. lr: 0.00005
  10. lr_decay: 0.90
  11. ## Loss functions
  12. margin: 0.1
  13. scale: 30
  14. nClasses: 774
  15. ## Load and save
  16. save_path: exps/resnet18_pre4
  17. ## Training and test data
  18. train_path: /root/public_storage/face_trainer/data/train
  19. test_path: /root/public_storage/face_trainer/data/valid
  20. test_list: /root/public_storage/face_trainer/data/valid_list.csv
  21. ## Model definition
  22. model: ResNet18
  23. # initial_model: models/res18_vggface1_baseline.model