Thanks!
On this topic does anyone know if there is discussion of adding a something like torchsummary to the pytorch code base? I’ve seen others ask on the forums but haven’t seen any definitive response or reason for its absence.