BaderLab / saber

Saber is a deep-learning based tool for information extraction in the biomedical domain. Pull requests are welcome! Note: this is a work in progress. Many things are broken, and the codebase is not stable.
https://baderlab.github.io/saber/
MIT License
102 stars 17 forks source link

Need nicely formatted string representation of a PyTorch model #140

Open JohnGiorgi opened 5 years ago

JohnGiorgi commented 5 years ago

Need to add a way to print out a nicely formatted representation of a PyTorch model. Wrap this under a summary() function in the BasePyTorchModel so that its usage is the same as the Keras models in Saber.

See here for inspiration.