I am trying to implement this paper : https://papers.nips.cc/paper/7454-collaborative-learning-for-deep-neural-networks.pdf 8
The proposed model in the paper has multiple classifier heads with shared layers between multiple classifiers, so the backward pass becomes a many to one graph.
How do I go forward with implementing such kind of backward() function ?