paddlepaddle-加载预训练模型

首先上官方例子

参考：

https://github.com/PaddlePaddle/models/blob/develop/PaddleCV/image_classification/train.py

place = fluid.CUDAPlace(0) if args.use_gpu else fluid.CPUPlace()
exe = fluid.Executor(place)
exe.run(startup_prog)

if checkpoint is not None:
    fluid.io.load_persistables(exe, checkpoint, main_program=train_prog)

add_arg('pretrained_model', str, None, "Whether to use pretrained model.")

pretrained_model = args.pretrained_model

if pretrained_model:
    def if_exist(var):
        return os.path.exists(os.path.join(pretrained_model, var.name))
    fluid.io.load_vars(exe, pretrained_model, main_program=train_prog, predicate=if_exist)