Model summary in PyTorch, based off of the original torchsummary.
conda install conda-forge::torchinfo