-
Notifications
You must be signed in to change notification settings - Fork 357
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
How to save tw.draw_model to image file #53
Comments
The result of tw.draw_model is a object of DotWrapper Calss. You can save tw.draw_model to image file as follow: img = tw.draw_model(model, [1, 3, 224, 224]) # list cantains the size of input tensor |
I find the method help info by help(tw.draw_model) using img.save, the png resolution is too low to see the model graph |
I can only run my model on server, so I can't see the model graph directly from the example code:
tw.draw_model.
So, anyone can tell me how to save the graph data to image file? Thanks.
The text was updated successfully, but these errors were encountered: