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()
使用pycaffe进行网络训练
最新推荐文章于 2024-05-12 23:14:58 发布