graphgen.py 文件源码

def __init__(self):
        self.seen = set()
        self.dot = pydot.Dot("resnet", graph_type="digraph", rankdir="TB")
        self.style_params = {"shape": "octagon",
                             "fillcolor": "gray",
                             "style": "filled",
                             "label": "",
                             "color": "none"}
        self.style_layers = {"shape": "box",
                             "fillcolor": "blue",
                             "style": "filled",
                             "label": "",
                             "color": "none"}