search

longcw / pytorch2caffe

Convert PyTorch model to Caffemodel
541 stars 151 forks source link

'Dot' object is not iterable #14

Closed rabbithui closed 6 years ago

rabbithui commented 6 years ago

I run "demo.py" ,and got the error message as follows: Traceback (most recent call last): File "/home/software/pycharm/helpers/pydev/pydevd.py", line 1580, in globals = debugger.run(setup['file'], None, None, is_module) File "/home/software/pycharm/helpers/pydev/pydevd.py", line 964, in run pydev_imports.execfile(file, globals, locals) # execute the script File "/home/RNN/crnn.pytorch-master_demo/demo_caffe.py", line 50, in plot_graph(output_var, os.path.join(caffemodel_dir, 'pytorch_graph.dot')) File "/home/RNN/crnn.pytorch-master_demo/pytorch2caffe.py", line 380, in plot_graph (graph,) = pydot.graph_from_dot_file(fname) TypeError: 'Dot' object is not iterable

hkchengrex commented 6 years ago

For reference, I solved it by installing a newer version of pydot:

sudo pip install -I pydot==1.2.4