使用pycaffe进行网络训练

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/zhongqianli/article/details/86139490

import caffe

caffe.set_device(0)
caffe.set_mode_gpu()
solver = caffe.SGDSolver("solver.prototxt")
# print layer shape
for k, v in solver.net.blobs.items():
	print(k, v.data.shape)
solver.solve()