GitHub repository of the related (inplace) paper. WIP
This is a toolbox to reproduce the various experiments conducted in the paper. Mainly, the engine subsection provide several useful function to train/compare/observe different types of networks.
A code for network comparaison can be found here.
Gradient Extension performed in the paper is demonstrated here.
You can watch CKA application to neural networks here.