Model summary in PyTorch similar to `model.summary()` in Keras
Why do you think that https://github.com/horovod/horovod is a good alternative to pytorch-summary
Model summary in PyTorch similar to `model.summary()` in Keras
Why do you think that https://github.com/horovod/horovod is a good alternative to pytorch-summary