Updated 6 months ago

py-torchsummary • Rank 22.6 • Science 23%

Model summary in PyTorch similar to `model.summary()` in Keras