请问caffe2怎么做迁移学习

大家好,我是一个caffe2的新手,我想请教一下caffe2如何做迁移学习,比如就是vgg16模型来说,我想只利用特征抽取的层,但是我试了好几天,当我load完init_net.pb和predict_net.pb后应该怎么做才能实现我说的只保留特征抽取的基层,把最后的几层去掉。我看了一些文章发现,貌似要直接利用一些类似hard code的方式,比如init_net.Proto().op.pop(20)什么的这类方法,请问这个是正确的方法么,是否还有其他方法来调整网络。
 
还有caffe2是否可以这样做,比如我先用model.fc创建了一个fc,然后我可以主动把预训练模型里对应名字的参数直接提取出来赋值给这个model.fc
已邀请:

要回复问题请先登录注册