waleedka / hiddenlayer

Neural network graphs and training metrics for PyTorch, Tensorflow, and Keras.

MIT License

1.79k stars 266 forks source link

module 'torch.jit' has no attribute '_get_trace_graph' #79

Closed zhongqiu1245 closed 4 years ago

zhongqiu1245 commented 4 years ago

Hi，thank you for your amazing job I try to use your work to visualize my own model When I run :

my model

import hiddenlayer as hl hl.build_graph(model, torch.zeros([1, 3, 224, 224])) but I got: AttributeError: module 'torch.jit' has no attribute '_get_trace_graph' Could you tell me how can I slove it? Thank you

ischihmo commented 4 years ago

https://github.com/microsoft/tensorwatch/issues/29#issuecomment-596304208

benihime91 commented 4 years ago

I am getting the same error.

austinmw commented 4 years ago

@zhongqiu1245 Hi, how did you solve this?