Yep, T. (2020). torchinfo [Computer software]. https://github.com/TylerYep/torchinfo