The steps defined in (REF ), (REF ), (REF ), (REF ) are repeated for the \(t\)  iterations of the routing algorithm. In order to train the network, Sabour et al. [1]} replaced the cross entropy loss with the margin loss.
