pytorch-summary
Model summary in PyTorch similar to model.summary() in Keras
4k stars
37 watching
412 forks
Language: Python
last commit: 7 months ago deep-learningkeraspytorchsummary
pytorch-summary Model summary in PyTorch similar to model.summary() in Keras