Lightweight reimplementation of netdissect for pytorch (under development) - View it on GitHub