tensorwatch
tensorwatch copied to clipboard
Problem in model_draw
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 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
.
I met the same problem. Have anyone solved this?
I met the same same problem. Is there anybody who can help us?
same issue.