###############################################################################################
‘’‘
tw.draw_model(model, [1, 3, 512, 512],png_filename='unet.png')
File "C:\Anaconda3\envs\pytorch1d10d2\lib\site-packages\tensorwatch__init__.py", line 35, in draw_model
g = pytorch_draw_model.draw_graph(model, input_shape)
File "C:\Anaconda3\envs\pytorch1d10d2\lib\site-packages\tensorwatch\model_graph\hiddenlayer\pytorch_draw_model.py", line 35, in draw_graph
dot = draw_img_classifier(model, args)
File "C:\Anaconda3\envs\pytorch1d10d2\lib\site-packages\tensorwatch\model_graph\hiddenlayer\pytorch_draw_model.py", line 63, in draw_img_classifier
g = SummaryGraph(non_para_model, dummy_input)
File "C:\Anaconda3\envs\pytorch1d10d2\lib\site-packages\tensorwatch\model_graph\hiddenlayer\summary_graph.py", line 221, in init
new_op['attrs'] = OrderedDict([(attr_name, node[attr_name]) for attr_name in node.attributeNames()])
File "C:\Anaconda3\envs\pytorch1d10d2\lib\site-packages\tensorwatch\model_graph\hiddenlayer\summary_graph.py", line 221, in
new_op['attrs'] = OrderedDict([(attr_name, node[attr_name]) for attr_name in node.attributeNames()])
File "C:\Anaconda3\envs\pytorch1d10d2\lib\site-packages\torch\onnx\utils.py", line 1232, in _node_getitem
return getattr(self, sel)(k)
AttributeError: 'torch._C.Node' object has no attribute 'ival'
Read This First
Remember: if we cannot reproduce your problem, we cannot find solution!
### OS Version=win10 64bit python Version= 3.9.7 Pytorch Version= 1.10.2 tensorwatch=0.9.1
############################################################################################### ‘’‘ tw.draw_model(model, [1, 3, 512, 512],png_filename='unet.png') File "C:\Anaconda3\envs\pytorch1d10d2\lib\site-packages\tensorwatch__init__.py", line 35, in draw_model g = pytorch_draw_model.draw_graph(model, input_shape) File "C:\Anaconda3\envs\pytorch1d10d2\lib\site-packages\tensorwatch\model_graph\hiddenlayer\pytorch_draw_model.py", line 35, in draw_graph
dot = draw_img_classifier(model, args) File "C:\Anaconda3\envs\pytorch1d10d2\lib\site-packages\tensorwatch\model_graph\hiddenlayer\pytorch_draw_model.py", line 63, in draw_img_classifier g = SummaryGraph(non_para_model, dummy_input) File "C:\Anaconda3\envs\pytorch1d10d2\lib\site-packages\tensorwatch\model_graph\hiddenlayer\summary_graph.py", line 221, in init new_op['attrs'] = OrderedDict([(attr_name, node[attr_name]) for attr_name in node.attributeNames()]) File "C:\Anaconda3\envs\pytorch1d10d2\lib\site-packages\tensorwatch\model_graph\hiddenlayer\summary_graph.py", line 221, in
new_op['attrs'] = OrderedDict([(attr_name, node[attr_name]) for attr_name in node.attributeNames()]) File "C:\Anaconda3\envs\pytorch1d10d2\lib\site-packages\torch\onnx\utils.py", line 1232, in _node_getitem return getattr(self, sel)(k) AttributeError: 'torch._C.Node' object has no attribute 'ival'
’‘’ ###############################################################################################
What's better than filing issue? Filing a pull request :).
------------------------------------ (Remove above before filing the issue) ------------------------------------