tensorwatch icon indicating copy to clipboard operation
tensorwatch copied to clipboard

Problem in model_draw

Open wxyz1234 opened this issue 4 years ago • 5 comments

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 avatar Mar 28 '20 03:03 wxyz1234

PNG

wxyz1234 avatar Mar 28 '20 03:03 wxyz1234

@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.

taosean avatar Apr 29 '20 10:04 taosean

I met the same problem. Have anyone solved this?

mengdong824 avatar Jul 03 '20 05:07 mengdong824

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

StrugglingForBetter avatar Jul 09 '20 09:07 StrugglingForBetter

same issue.

haibara-ai avatar Aug 03 '20 02:08 haibara-ai