Caffe转移参数子集

时间:2015-12-07 10:55:33

标签: python machine-learning neural-network deep-learning caffe

我想从受过训练的网络 n 中取出给定图层中 A 参数的子集,并将此子集转移到另一个网络 B 的图层。 在转移此子集的层中,应该随机初始化所述层(总转移子集)的其余参数。

这可以实现吗? (我知道可以用整层参数完成)

谢谢!

1 个答案:

答案 0 :(得分:2)

这可以在python中完成。见net surgery tutorial
基本上,您应该实例化两个caffe.Net对象,其中一个参数为A,另一个参数为随机。然后,您只需将参数子集从A分配到B,然后保存B即可完成。