search

PaddlePaddle / PaddleOCR

Awesome multilingual OCR toolkits based on PaddlePaddle (practical ultra lightweight OCR system, support 80+ languages recognition, provide data annotation and synthesis tools, support training and deployment among server, mobile, embedded and IoT devices)
Apache License 2.0
40.26k stars 7.45k forks source link

PPOCR V3使用dml训练模型，一直显示pretrained param not in model #11282

Open Double97828 opened 7 months ago

Double97828 commented 7 months ago

请提供下述完整信息以便快速定位问题/Please provide the following information to quickly locate the problem

我们提供了AceIssueSolver来帮助你解答问题，你是否想要它来解答(请填写yes/no)?/We provide AceIssueSolver to solve issues, do you want it? (Please write yes/no):

请尽量不要包含图片在问题中/Please try to not include the image in the issue.

我的yaml文件是这样的，每次做训练都会报warning， Global: use_gpu: true epoch_num: 300 log_smooth_window: 20 print_batch_step: 2 save_model_dir: ./output/invoice/ch_PP-OCR_dml save_epoch_step: 1200

eval_batch_step: [0, 50] cal_metric_during_train: False pretrained_model: ./pretrain_models/ResNet200_vd_pretrained checkpoints:
save_inference_dir: use_visualdl: False infer_img: save_res_path:

Architecture: name: DistillationModel algorithm: Distillation model_type: det Models: Student: return_all_feats: false model_type: det algorithm: DB Backbone: name: ResNet_vd in_channels: 3 layers: 50 Neck: name: LKPAN out_channels: 256 Head: name: DBHead kernel_list: [7,2,2] k: 50 Student2: return_all_feats: false model_type: det algorithm: DB Backbone: name: ResNet_vd in_channels: 3 layers: 50 Neck: name: LKPAN out_channels: 256 Head: name: DBHead kernel_list: [7,2,2] k: 50

Loss: name: CombinedLoss loss_config_list:

Optimizer: name: Adam beta1: 0.9 beta2: 0.999 lr: name: Cosine learning_rate: 0.001 warmup_epoch: 2 regularizer: name: 'L2' factor: 0

PostProcess: name: DistillationDBPostProcess model_name: ["Student", "Student2"] key: head_out thresh: 0.3 box_thresh: 0.6 max_candidates: 1000 unclip_ratio: 1.5

Metric: name: DistillationMetric base_metric_name: DetMetric main_indicator: hmean key: "Student"

Train: dataset: name: SimpleDataSet data_dir: ./train_data/invoice/ label_file_list:

Eval: dataset: name: SimpleDataSet data_dir: ./train_data/invoice/ label_file_list:

Haeppypuppy commented 4 months ago

同問