Commit 2da8600
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
@@ -142,7 +142,7 @@ def get_pydot_graph(caffe_net, rankdir, label_edges=True): | |||
| 142 | 142 | ------- | |
| 143 | 143 | pydot graph object | |
| 144 | 144 | """ | |
| 145 | - pydot_graph = pydot.Dot(caffe_net.name, | ||
| 145 | + pydot_graph = pydot.Dot(caffe_net.name if caffe_net.name else 'Net', | ||
| 146 | 146 | graph_type='digraph', | |
| 147 | 147 | rankdir=rankdir) | |
| 148 | 148 | pydot_nodes = {} | |
0 commit comments