add checkpoints

This commit is contained in:
tink2123 2020-05-12 20:05:13 +08:00
parent d24a39df88
commit ec6d730c97
1 changed files with 3 additions and 3 deletions

View File

@ -5,8 +5,8 @@ Global:
log_smooth_window: 20 log_smooth_window: 20
print_batch_step: 10 print_batch_step: 10
save_model_dir: output_ic15 save_model_dir: output_ic15
save_epoch_step: 300 save_epoch_step: 3
eval_batch_step: 200 eval_batch_step: 2000
train_batch_size_per_card: 256 train_batch_size_per_card: 256
test_batch_size_per_card: 256 test_batch_size_per_card: 256
image_shape: [3, 32, 100] image_shape: [3, 32, 100]
@ -16,7 +16,7 @@ Global:
loss_type: ctc loss_type: ctc
reader_yml: ./configs/rec/rec_icdar15_reader.yml reader_yml: ./configs/rec/rec_icdar15_reader.yml
pretrain_weights: ./pretrain_models/CRNN/best_accuracy pretrain_weights: ./pretrain_models/CRNN/best_accuracy
checkpoints:
Architecture: Architecture:
function: ppocr.modeling.architectures.rec_model,RecModel function: ppocr.modeling.architectures.rec_model,RecModel