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

Problem in model_draw #60

Open
wxyz1234 opened this issue Mar 28, 2020 · 5 comments
Open

Problem in model_draw #60

wxyz1234 opened this issue Mar 28, 2020 · 5 comments

Comments

@wxyz1234
Copy link

wxyz1234 commented Mar 28, 2020

PNG

When I use
alexnet_model = torchvision.models.alexnet()
tw.draw_model(alexnet_model, [1, 3, 224, 224])
there is a strange output,What's wrong with it?
OS version:Win10
Python version:3.6.10
Pytorch version:1.3.1

@wxyz1234
Copy link
Author

PNG

@taosean
Copy link

taosean commented Apr 29, 2020

@wxyz1234
Hi, I have the same issue as you did. Have you solve this?

Besides, I found that the alexnet model drawed in README and network_arch.ipynb look different.
It's pretty weird.

According to this article, I changed the version of tensorwatch to 0.8.7 (I was using 0.9.1) and do the mentioned modification to site-packages\tensorwatch\model_graph\hiddenlayer\pytorch_draw_model.py, now I can get the same drawing as in network_arch.ipynb.

However, I still cannot get the drawing in README.

@mengdong824
Copy link

I met the same problem. Have anyone solved this?

@StrugglingForBetter
Copy link

I met the same same problem. Is there anybody who can help us?

@haibara-ai
Copy link

same issue.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

6 participants
@haibara-ai @taosean @StrugglingForBetter @mengdong824 @wxyz1234 and others