文字图片方向分类训练时如何使用预训练模型text_image_orientation_pretrained？

[tianwenzhe]

请问，文字图片方向分类训练时如何使用预训练模型text_image_orientation_pretrained？

直接将PPLCNet_x1_0.yaml的Global中pretrained_model设置为text_image_orientation_pretrained.pdparams的路径（text_image_orientation_pretrained不加后缀pdparams）即可吗？
如：
Global:
checkpoints: null

pretrained_model: null

pretrained_model: pretrained_model/text_image_orientation/text_image_orientation_pretrained
output_dir: ./output/
device: gpu
save_interval: 1
eval_during_train: True
eval_interval: 1

epochs: 60

epochs: 1
print_batch_step: 10
use_visualdl: False

used for static mode and model export

image_shape: [3, 224, 224]
save_inference_dir: ./inference

[tianwenzhe]

我尝试了，这样是正确的

[TingquanGao]

嗯嗯，这样可以的。