microsoft / tensorwatch

Debugging, monitoring and visualization for Python Machine Learning and Data Science
MIT License
3.42k stars 362 forks source link

problem about draw_model() #70

Open Huangdebo opened 4 years ago

Huangdebo commented 4 years ago

When I run:

import tensorwatch as tw import torchvision.models alexnet_model = torchvision.models.alexnet() tw.model_stats(alexnet_model, [1, 3, 224, 224])

there is a strange output,What's wrong with it?

OS version:Win7 Python version:3.8.3 Pytorch version:1.5.1

image

CplusBest commented 4 years ago

I met the problem too. And someone gave a list of requirement(low version): graphviz = 2.38 pytorch = 1.2 tensorwatch = 0.8.7 torchvision = 0.4.0 But i do not want fallback version especially pytorch. =.=

Huangdebo commented 4 years ago

I met the problem too. And someone gave a list of requirement(low version): graphviz = 2.38 pytorch = 1.2 tensorwatch = 0.8.7 torchvision = 0.4.0 But i do not want fallback version especially pytorch. =.=

thank you ! i don't want to fallback the version too, just waitting for the new version of tensorwatch that supports pytorch-1.5.1