TODOs. A more user-friendly warning for un-defined modules. [Done] Supports for models in torchvision (e.g., residual add). Layer wise printing Integration with torchprofile?