draw_net: accept prototxt without name

Fixes #3819

diff --git a/python/caffe/draw.py b/python/caffe/draw.py
index cfa3fc5..61205ca 100644 (file)
--- a/python/caffe/draw.py
+++ b/python/caffe/draw.py
@@ -142,7 +142,7 @@ def get_pydot_graph(caffe_net, rankdir, label_edges=True):
     -------
     pydot graph object
     """
-    pydot_graph = pydot.Dot(caffe_net.name,
+    pydot_graph = pydot.Dot(caffe_net.name if caffe_net.name else 'Net',
                             graph_type='digraph',
                             rankdir=rankdir)
     pydot_nodes = {}