Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Need nicely formatted string representation of a PyTorch model #140

Open
JohnGiorgi opened this issue Jun 3, 2019 · 0 comments
Open

Need nicely formatted string representation of a PyTorch model #140

JohnGiorgi opened this issue Jun 3, 2019 · 0 comments
Assignees
Labels

Comments

@JohnGiorgi
Copy link
Contributor

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.

@JohnGiorgi JohnGiorgi added enhancement New feature or request design production labels Jun 3, 2019
@JohnGiorgi JohnGiorgi self-assigned this Jun 3, 2019
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Projects
None yet
Development

No branches or pull requests

1 participant