预训练模型
Created by: wbchief
您好,请问在PaddlePaddle下预训练的VGG16等模型,如何使用这些预训练好的模型来更新自己利用PaddlePaddle构建的动态图模型网络的参数。就如Pytorch中的这段代码:
mod = models.vgg16(pretrained = True) fs = self.frontend.state_dict() ms = mod.state_dict() for key in fs: fs[key] = ms['features.'+key] self.frontend.load_state_dict(fs)
谢谢!在网上没能找到相关资料。