Почему результаты Pycaffe отстали всегда 0？

Question

net.blobs['data'].data[...] = transformed_image
output = net.forward()

output_prob = output['prob'][0]  # the output probability vector for the 
first image in the batch

print 'predicted class is:', output_prob.argmax()

label_index = output_prob.argmax()
caffeLabel = np.zeros((1,1000))
caffeLabel[0,label_index] = 1;

vis_layer = 'pool5' # visualization layer

grads=net.backward(diffs=[vis_layer],**{'prob':caffeLabel})
print(np.sum(grads))

Я хочу получить градиенты таким способом, но print (np.sum (grads)) всегда равен 0, я изменяю слой conv5 или другие слои, он не работал!

Answer 1

Я решил проблему, добавил следующий код в 'deploy.prototxt'

force_backward:true