1234567891011121314151617181920212223242526272829 |
- ## Data loader
- batch_size: 200
- ## Training details
- max_epoch: 15
- trainfunc: softmax
- ## Optimizer
- scheduler: steplr
- optimizer: adam
- lr: 0.00005
- lr_decay: 0.90
- ## Loss functions
- margin: 0.1
- scale: 30
- nClasses: 774
- ## Load and save
- save_path: exps/resnet18_pre4
- ## Training and test data
- train_path: /root/public_storage/face_trainer/data/train
- test_path: /root/public_storage/face_trainer/data/valid
- test_list: /root/public_storage/face_trainer/data/valid_list.csv
- ## Model definition
- model: ResNet18
- # initial_model: models/res18_vggface1_baseline.model
|