capsule-networks
capsule-networks copied to clipboard
Why should stop the gradient of u_hat?
"u_hat_stopped = tf.stop_gradient(u_hat, name='stop_gradient')"
I don't think W can be updated in that case