Description
import tensorwatch as tw
import torchvision.models
alexnet_model = torchvision.models.alexnet()
tw.draw_model(alexnet_model, [1, 3, 224, 224])
**---------------------------------------------------------------------------
AttributeError Traceback (most recent call last)
in
1 alexnet_model = torchvision.models.alexnet()
----> 2 tw.draw_model(alexnet_model, [1, 3, 224, 224])
E:\anaconda3\envs\torch_tensorwatch\lib\site-packages\tensorwatch_init_.py in draw_model(model, input_shape, orientation, png_filename)
33 def draw_model(model, input_shape=None, orientation='TB'): #orientation = 'LR' for landscpe
34 from .model_graph.hiddenlayer import graph
---> 35 g = graph.build_graph(model, input_shape, orientation=orientation)
36 return g
37
E:\anaconda3\envs\torch_tensorwatch\lib\site-packages\tensorwatch\model_graph\hiddenlayer\pytorch_draw_model.py in draw_graph(model, args)
E:\anaconda3\envs\torch_tensorwatch\lib\site-packages\tensorwatch\model_graph\hiddenlayer\pytorch_draw_model.py in draw_img_classifier(model, dataset, display_param_nodes, rankdir, styles, input_shape)
E:\anaconda3\envs\torch_tensorwatch\lib\site-packages\tensorwatch\model_graph\hiddenlayer\summary_graph.py in init(self, model, dummy_input, apply_scope_name_workarounds)
E:\anaconda3\envs\torch_tensorwatch\lib\site-packages\tensorwatch\model_graph\hiddenlayer\summary_graph.py in __add_param(self, n)
AttributeError: 'torch._C.Value' object has no attribute 'debugName'**
python 3.6
torch 1.1.0
torchvision 0.3.0
tensorwatch 0.9.0
I just use the simple example with the jupyter notebook , but it raise the error which confused me . I will be appreciate it if anyone can reply me with the solution to the problem.