Caffe 传递参数子集
Caffe transfer subset of parameters
我想从训练好的网络A
[=24中取出给定层中n
参数的一个子集=],并将这个子集转移到另一个网络的一层B
。
在这个子集被转移到的层中,该层的剩余参数(总 - 转移子集)应该被随机初始化。
这可以实现吗?
(我知道可以用整层参数来完成)
谢谢!
这可以在 python 中完成。参见 net surgery tutorial。
基本上,您应该实例化两个 caffe.Net
对象,一个参数为 A
,另一个参数为随机。然后,您只需将 A
的参数子集分配给 B
,保存 B
即可。
我想从训练好的网络A
[=24中取出给定层中n
参数的一个子集=],并将这个子集转移到另一个网络的一层B
。
在这个子集被转移到的层中,该层的剩余参数(总 - 转移子集)应该被随机初始化。
这可以实现吗? (我知道可以用整层参数来完成)
谢谢!
这可以在 python 中完成。参见 net surgery tutorial。
基本上,您应该实例化两个 caffe.Net
对象,一个参数为 A
,另一个参数为随机。然后,您只需将 A
的参数子集分配给 B
,保存 B
即可。