I\'m using the subclassing API to build a simple conv net and I want to use the summary method to get an idea of what the architecture looks like for my model. However, when
