预训练模型参数问题
Created by: wbchief
您好!我现在需要在自己搭建的动态图模型中用到PaddleHub中的VGG16模型中的模型参数,我应该如何获取到预训练的模型参数并在放到我的模型中使用呢?
现在我使用 import paddlehub as hub module = hub.Module(name="vgg16_imagenet") input_dict, output_dict, program = module.context(trainable=True)
import paddle.fluid as fluid with fluid.program_guard(program): param_list = fluid.framework.default_main_program().block(0).all_parameters() print(fluid.default_main_program().num_blocks) print(param_list) 这段程序获取到模型参数,但是这样输出只有类似于 [name: "@HUB_vgg16_imagenet@fc8_offset" type { type: LOD_TENSOR lod_tensor { tensor { data_type: FP32 dims: 1000 } } } persistable: true 的结构,并没有看到模型参数w和b, 我该如何利用Variable的set_value()方法给另一个模型设置预训练的VGG16模型的参数呢?
小建议: 有关动态图方面的文档不是很多,资料很难找到。 期待您的回答。谢谢!