notebook上使用py-caffe训练调试模型

    最近在使用caffe的prototxt设计一个新网络，写了一些骚操作，自己也不是很确定计算过程是否符合预期，所以想在训练若干步之后抽出我设计的那几层，观察一下输出值。直接在代码或日志层来检查指定层的输出值当然也是可以的，但是不直观，也不灵活，于是想到了py-caffe结合notebook的方式。查阅了一下发现还真可以直接在notebook调用py-caffe的接口跑训练，操作也异常简单，嘿嘿！

import caffe

#指定好solver.prototxt以及训练所使用的优化器
solver_path = "solver.prototxt"
solver = caffe.AdamSolver(solver_path)

#也可以加载预训练好的模型
weight_path = "FACE-REG-GroupNetV4dResK16_112x112_LS3D-W-DAV3-190420_iter_654000.caffemodel"
solver.net.copy_from(weight_path)

#训练跑几步
solver.step(1)  #参数表示迭代次数
net = solver.net  

#查看各层blob数值及参数信息了
for k,v in net.blobs.items():
    print(k,v.data.shape)

for k,v in net.params.items():
    print(k, v[0].data.shape)

#你也可以指定查看某层的数据或参数,so easy!
net.blobs['fc7'].data[0]