pytorch-model-summary · PyPI
https://pypi.org/project/pytorch-model-summary30.08.2020 · Pytorch Model Summary -- Keras style model.summary() for PyTorch. It is a Keras style model.summary() implementation for PyTorch. This is an Improved PyTorch library of modelsummary. Like in modelsummary, It does not care with number of Input parameter! Improvements: For user defined pytorch layers, now summary can show layers inside it
pytorch-model-summary · PyPI
pypi.org › project › pytorch-model-summaryAug 30, 2020 · Pytorch Model Summary -- Keras style model.summary() for PyTorch. It is a Keras style model.summary() implementation for PyTorch. This is an Improved PyTorch library of modelsummary. Like in modelsummary, It does not care with number of Input parameter! Improvements: For user defined pytorch layers, now summary can show layers inside it