pepy.tech

torchinfo

1.8.0
9.92M

Model summary in PyTorch, based off of the original torchsummary.