Skip to content

Commit b86b0ae

Browse files
committed
Merge pull request BVLC#4071 from mnogu/optional-name
draw_net: accept prototxt without name
2 parents f623d04 + 2da8600 commit b86b0ae

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

python/caffe/draw.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -142,7 +142,7 @@ def get_pydot_graph(caffe_net, rankdir, label_edges=True):
142142
-------
143143
pydot graph object
144144
"""
145-
pydot_graph = pydot.Dot(caffe_net.name,
145+
pydot_graph = pydot.Dot(caffe_net.name if caffe_net.name else 'Net',
146146
graph_type='digraph',
147147
rankdir=rankdir)
148148
pydot_nodes = {}

0 commit comments

Comments
 (0)