Skip to content

Instantly share code, notes, and snippets.

@qizhuli

qizhuli/pytorchviz_output.dot

Created February 28, 2019 16:30
Show Gist options
  • Save qizhuli/0b6bca77aea1778d4ccc30660393f491 to your computer and use it in GitHub Desktop.
Save qizhuli/0b6bca77aea1778d4ccc30660393f491 to your computer and use it in GitHub Desktop.
Download ZIP
Output of vis_model.py of "python tools/vis_model.py --config-file configs/e2e_mask_rcnn_R_50_FPN_1x.yaml"
Raw
pytorchviz_output.dot
digraph {
graph [size="145.79999999999998,145.79999999999998"]
node [align=left fontsize=12 height=0.2 ranksep=0.1 shape=box style=filled]
140156541355456 [label=NllLossBackward fillcolor=darkolivegreen1]
140156541355512 -> 140156541355456
140156541355512 [label=LogSoftmaxBackward]
140156541355904 -> 140156541355512
140156541355904 [label=AddmmBackward]
140156541355232 -> 140156541355904
140156541355232 [label="roi_heads.box.predictor.cls_score.bias
(81)" fillcolor=lightblue]
140156541355680 -> 140156541355904
140156541355680 [label=ReluBackward0]
140156541355792 -> 140156541355680
140156541355792 [label=AddmmBackward]
140156541355288 -> 140156541355792
140156541355288 [label="roi_heads.box.feature_extractor.fc7.bias
(1024)" fillcolor=lightblue]
140156541355064 -> 140156541355792
140156541355064 [label=ReluBackward0]
140156541356016 -> 140156541355064
140156541356016 [label=AddmmBackward]
140156541356184 -> 140156541356016
140156541356184 [label="roi_heads.box.feature_extractor.fc6.bias
(1024)" fillcolor=lightblue]
140156541356296 -> 140156541356016
140156541356296 [label=ViewBackward]
140156541356464 -> 140156541356296
140156541356464 [label=IndexPutBackward]
140156541356576 -> 140156541356464
140156541356576 [label=IndexPutBackward]
140156541356688 -> 140156541356576
140156541356688 [label=IndexPutBackward]
140156541356800 -> 140156541356688
140156541356800 [label=IndexPutBackward]
93906381257640 -> 140156541356800
93906381257640 [label=_ROIAlignBackward]
140156541356968 -> 93906381257640
140156541356968 [label=MkldnnConvolutionBackward]
140156541357080 -> 140156541356968
140156541357080 [label=AddBackward0]
140156541357304 -> 140156541357080
140156541357304 [label=MkldnnConvolutionBackward]
140156541357472 -> 140156541357304
140156541357472 [label=ReluBackward1]
140156541357696 -> 140156541357472
140156541357696 [label=AddBackward0]
140156541357808 -> 140156541357696
140156541357808 [label=AddBackward0]
140156541357976 -> 140156541357808
140156541357976 [label=MulBackward0]
140156541452360 -> 140156541357976
140156541452360 [label=MkldnnConvolutionBackward]
140156541452472 -> 140156541452360
140156541452472 [label=ReluBackward1]
140156541452584 -> 140156541452472
140156541452584 [label=AddBackward0]
140156541452696 -> 140156541452584
140156541452696 [label=MulBackward0]
140156541452808 -> 140156541452696
140156541452808 [label=MkldnnConvolutionBackward]
140156541452920 -> 140156541452808
140156541452920 [label=ReluBackward1]
140156541453032 -> 140156541452920
140156541453032 [label=AddBackward0]
140156541453144 -> 140156541453032
140156541453144 [label=MulBackward0]
140156541453256 -> 140156541453144
140156541453256 [label=MkldnnConvolutionBackward]
140156541357864 -> 140156541453256
140156541357864 [label=ReluBackward1]
140156541453424 -> 140156541357864
140156541453424 [label=AddBackward0]
140156541453536 -> 140156541453424
140156541453536 [label=AddBackward0]
140156541453704 -> 140156541453536
140156541453704 [label=MulBackward0]
140156541453816 -> 140156541453704
140156541453816 [label=MkldnnConvolutionBackward]
140156541453928 -> 140156541453816
140156541453928 [label=ReluBackward1]
140156541454040 -> 140156541453928
140156541454040 [label=AddBackward0]
140156541454152 -> 140156541454040
140156541454152 [label=MulBackward0]
140156541454264 -> 140156541454152
140156541454264 [label=MkldnnConvolutionBackward]
140156541454376 -> 140156541454264
140156541454376 [label=ReluBackward1]
140156541454488 -> 140156541454376
140156541454488 [label=AddBackward0]
140156541454600 -> 140156541454488
140156541454600 [label=MulBackward0]
140156541454712 -> 140156541454600
140156541454712 [label=MkldnnConvolutionBackward]
140156541453592 -> 140156541454712
140156541453592 [label=ReluBackward1]
140156541454880 -> 140156541453592
140156541454880 [label=AddBackward0]
140156541454992 -> 140156541454880
140156541454992 [label=AddBackward0]
140156541455160 -> 140156541454992
140156541455160 [label=MulBackward0]
140156541455272 -> 140156541455160
140156541455272 [label=MkldnnConvolutionBackward]
140156541455384 -> 140156541455272
140156541455384 [label=ReluBackward1]
140156541455496 -> 140156541455384
140156541455496 [label=AddBackward0]
140156541455608 -> 140156541455496
140156541455608 [label=MulBackward0]
140156541455720 -> 140156541455608
140156541455720 [label=MkldnnConvolutionBackward]
140156541455832 -> 140156541455720
140156541455832 [label=ReluBackward1]
140156541455944 -> 140156541455832
140156541455944 [label=AddBackward0]
140156541456056 -> 140156541455944
140156541456056 [label=MulBackward0]
140156541456168 -> 140156541456056
140156541456168 [label=MkldnnConvolutionBackward]
140156541456280 -> 140156541456168
140156541456280 [label=MaxPool2DWithIndicesBackward]
140156541460552 -> 140156541456280
140156541460552 [label=ReluBackward1]
140156541460664 -> 140156541460552
140156541460664 [label=AddBackward0]
140156541460776 -> 140156541460664
140156541460776 [label=MulBackward0]
140156541460888 -> 140156541460776
140156541460888 [label=MkldnnConvolutionBackward]
140156541461000 -> 140156541460888
140156541461000 [label="dummy_image
(1, 3, 1600, 800)" fillcolor=lightblue]
140156541455048 -> 140156541454880
140156541455048 [label=AddBackward0]
140156541455216 -> 140156541455048
140156541455216 [label=MulBackward0]
140156541455440 -> 140156541455216
140156541455440 [label=MkldnnConvolutionBackward]
140156541456280 -> 140156541455440
140156541453592 -> 140156541453424
140156541357864 -> 140156541357696
140156541357528 -> 140156541357304
140156541357528 [label="backbone.fpn.fpn_inner1.weight
(256, 256, 1, 1)" fillcolor=lightblue]
140156541357584 -> 140156541357304
140156541357584 [label="backbone.fpn.fpn_inner1.bias
(256)" fillcolor=lightblue]
140156541357360 -> 140156541357080
140156541357360 [label=UpsampleNearest2DBackward]
140156541357640 -> 140156541357360
140156541357640 [label=AddBackward0]
140156541358032 -> 140156541357640
140156541358032 [label=MkldnnConvolutionBackward]
140156541452528 -> 140156541358032
140156541452528 [label=ReluBackward1]
140156541452976 -> 140156541452528
140156541452976 [label=AddBackward0]
140156541453200 -> 140156541452976
140156541453200 [label=AddBackward0]
140156541453480 -> 140156541453200
140156541453480 [label=MulBackward0]
140156541453760 -> 140156541453480
140156541453760 [label=MkldnnConvolutionBackward]
140156541453984 -> 140156541453760
140156541453984 [label=ReluBackward1]
140156541454320 -> 140156541453984
140156541454320 [label=AddBackward0]
140156541454544 -> 140156541454320
140156541454544 [label=MulBackward0]
140156541454768 -> 140156541454544
140156541454768 [label=MkldnnConvolutionBackward]
140156541454936 -> 140156541454768
140156541454936 [label=ReluBackward1]
140156541455552 -> 140156541454936
140156541455552 [label=AddBackward0]
140156541455776 -> 140156541455552
140156541455776 [label=MulBackward0]
140156541456000 -> 140156541455776
140156541456000 [label=MkldnnConvolutionBackward]
140156541453312 -> 140156541456000
140156541453312 [label=ReluBackward1]
140156541460608 -> 140156541453312
140156541460608 [label=AddBackward0]
140156541460832 -> 140156541460608
140156541460832 [label=AddBackward0]
140156541461168 -> 140156541460832
140156541461168 [label=MulBackward0]
140156541461112 -> 140156541461168
140156541461112 [label=MkldnnConvolutionBackward]
140156541461336 -> 140156541461112
140156541461336 [label=ReluBackward1]
140156541461504 -> 140156541461336
140156541461504 [label=AddBackward0]
140156541461616 -> 140156541461504
140156541461616 [label=MulBackward0]
140156541461728 -> 140156541461616
140156541461728 [label=MkldnnConvolutionBackward]
140156541461840 -> 140156541461728
140156541461840 [label=ReluBackward1]
140156541462008 -> 140156541461840
140156541462008 [label=AddBackward0]
140156541462120 -> 140156541462008
140156541462120 [label=MulBackward0]
140156541462232 -> 140156541462120
140156541462232 [label=MkldnnConvolutionBackward]
140156541460944 -> 140156541462232
140156541460944 [label=ReluBackward1]
140156541462456 -> 140156541460944
140156541462456 [label=AddBackward0]
140156541462568 -> 140156541462456
140156541462568 [label=AddBackward0]
140156541462736 -> 140156541462568
140156541462736 [label=MulBackward0]
140156541462848 -> 140156541462736
140156541462848 [label=MkldnnConvolutionBackward]
140156541462960 -> 140156541462848
140156541462960 [label=ReluBackward1]
140156541463128 -> 140156541462960
140156541463128 [label=AddBackward0]
140156541463240 -> 140156541463128
140156541463240 [label=MulBackward0]
140156541463352 -> 140156541463240
140156541463352 [label=MkldnnConvolutionBackward]
140156541463464 -> 140156541463352
140156541463464 [label=ReluBackward1]
140156541463632 -> 140156541463464
140156541463632 [label=AddBackward0]
140156541463744 -> 140156541463632
140156541463744 [label=MulBackward0]
140156541463856 -> 140156541463744
140156541463856 [label=MkldnnConvolutionBackward]
140156541462624 -> 140156541463856
140156541462624 [label=ReluBackward1]
140156541464080 -> 140156541462624
140156541464080 [label=AddBackward0]
140156541464192 -> 140156541464080
140156541464192 [label=AddBackward0]
140156541464360 -> 140156541464192
140156541464360 [label=MulBackward0]
140156541464472 -> 140156541464360
140156541464472 [label=MkldnnConvolutionBackward]
140156541476936 -> 140156541464472
140156541476936 [label=ReluBackward1]
140156541477104 -> 140156541476936
140156541477104 [label=AddBackward0]
140156541477216 -> 140156541477104
140156541477216 [label=MulBackward0]
140156541477328 -> 140156541477216
140156541477328 [label=MkldnnConvolutionBackward]
140156541477440 -> 140156541477328
140156541477440 [label=ReluBackward1]
140156541477608 -> 140156541477440
140156541477608 [label=AddBackward0]
140156541477720 -> 140156541477608
140156541477720 [label=MulBackward0]
140156541477832 -> 140156541477720
140156541477832 [label=MkldnnConvolutionBackward]
140156541357472 -> 140156541477832
140156541477944 -> 140156541477832
140156541477944 [label="backbone.body.layer2.0.conv1.weight
(128, 256, 1, 1)" fillcolor=lightblue]
140156541477496 -> 140156541477328
140156541477496 [label="backbone.body.layer2.0.conv2.weight
(128, 128, 3, 3)" fillcolor=lightblue]
140156541476992 -> 140156541464472
140156541476992 [label="backbone.body.layer2.0.conv3.weight
(512, 128, 1, 1)" fillcolor=lightblue]
140156541464248 -> 140156541464080
140156541464248 [label=AddBackward0]
140156541464416 -> 140156541464248
140156541464416 [label=MulBackward0]
140156541477048 -> 140156541464416
140156541477048 [label=MkldnnConvolutionBackward]
140156541357472 -> 140156541477048
140156541477384 -> 140156541477048
140156541477384 [label="backbone.body.layer2.0.downsample.0.weight
(512, 256, 1, 1)" fillcolor=lightblue]
140156541463968 -> 140156541463856
140156541463968 [label="backbone.body.layer2.1.conv1.weight
(128, 512, 1, 1)" fillcolor=lightblue]
140156541463520 -> 140156541463352
140156541463520 [label="backbone.body.layer2.1.conv2.weight
(128, 128, 3, 3)" fillcolor=lightblue]
140156541463016 -> 140156541462848
140156541463016 [label="backbone.body.layer2.1.conv3.weight
(512, 128, 1, 1)" fillcolor=lightblue]
140156541462624 -> 140156541462456
140156541462344 -> 140156541462232
140156541462344 [label="backbone.body.layer2.2.conv1.weight
(128, 512, 1, 1)" fillcolor=lightblue]
140156541461896 -> 140156541461728
140156541461896 [label="backbone.body.layer2.2.conv2.weight
(128, 128, 3, 3)" fillcolor=lightblue]
140156541461392 -> 140156541461112
140156541461392 [label="backbone.body.layer2.2.conv3.weight
(512, 128, 1, 1)" fillcolor=lightblue]
140156541460944 -> 140156541460608
140156541456224 -> 140156541456000
140156541456224 [label="backbone.body.layer2.3.conv1.weight
(128, 512, 1, 1)" fillcolor=lightblue]
140156541455104 -> 140156541454768
140156541455104 [label="backbone.body.layer2.3.conv2.weight
(128, 128, 3, 3)" fillcolor=lightblue]
140156541454096 -> 140156541453760
140156541454096 [label="backbone.body.layer2.3.conv3.weight
(512, 128, 1, 1)" fillcolor=lightblue]
140156541453312 -> 140156541452976
140156541452640 -> 140156541358032
140156541452640 [label="backbone.fpn.fpn_inner2.weight
(256, 512, 1, 1)" fillcolor=lightblue]
140156541452752 -> 140156541358032
140156541452752 [label="backbone.fpn.fpn_inner2.bias
(256)" fillcolor=lightblue]
140156541357752 -> 140156541357640
140156541357752 [label=UpsampleNearest2DBackward]
140156541452864 -> 140156541357752
140156541452864 [label=AddBackward0]
140156541453648 -> 140156541452864
140156541453648 [label=MkldnnConvolutionBackward]
140156541454208 -> 140156541453648
140156541454208 [label=ReluBackward1]
140156541455328 -> 140156541454208
140156541455328 [label=AddBackward0]
140156541456112 -> 140156541455328
140156541456112 [label=AddBackward0]
140156541461056 -> 140156541456112
140156541461056 [label=MulBackward0]
140156541460720 -> 140156541461056
140156541460720 [label=MkldnnConvolutionBackward]
140156541461448 -> 140156541460720
140156541461448 [label=ReluBackward1]
140156541461560 -> 140156541461448
140156541461560 [label=AddBackward0]
140156541462176 -> 140156541461560
140156541462176 [label=MulBackward0]
140156541462064 -> 140156541462176
140156541462064 [label=MkldnnConvolutionBackward]
140156541462680 -> 140156541462064
140156541462680 [label=ReluBackward1]
140156541462904 -> 140156541462680
140156541462904 [label=AddBackward0]
140156541463296 -> 140156541462904
140156541463296 [label=MulBackward0]
140156541463184 -> 140156541463296
140156541463184 [label=MkldnnConvolutionBackward]
140156541455664 -> 140156541463184
140156541455664 [label=ReluBackward1]
140156541463688 -> 140156541455664
140156541463688 [label=AddBackward0]
140156541464304 -> 140156541463688
140156541464304 [label=AddBackward0]
140156541477272 -> 140156541464304
140156541477272 [label=MulBackward0]
140156541477776 -> 140156541477272
140156541477776 [label=MkldnnConvolutionBackward]
140156541477552 -> 140156541477776
140156541477552 [label=ReluBackward1]
140156541478112 -> 140156541477552
140156541478112 [label=AddBackward0]
140156541478056 -> 140156541478112
140156541478056 [label=MulBackward0]
140156541478280 -> 140156541478056
140156541478280 [label=MkldnnConvolutionBackward]
140156541478392 -> 140156541478280
140156541478392 [label=ReluBackward1]
140156541478560 -> 140156541478392
140156541478560 [label=AddBackward0]
140156541478672 -> 140156541478560
140156541478672 [label=MulBackward0]
140156541478784 -> 140156541478672
140156541478784 [label=MkldnnConvolutionBackward]
140156541464528 -> 140156541478784
140156541464528 [label=ReluBackward1]
140156541479008 -> 140156541464528
140156541479008 [label=AddBackward0]
140156541479120 -> 140156541479008
140156541479120 [label=AddBackward0]
140156541479288 -> 140156541479120
140156541479288 [label=MulBackward0]
140156541479400 -> 140156541479288
140156541479400 [label=MkldnnConvolutionBackward]
140156541479512 -> 140156541479400
140156541479512 [label=ReluBackward1]
140156541479680 -> 140156541479512
140156541479680 [label=AddBackward0]
140156541479792 -> 140156541479680
140156541479792 [label=MulBackward0]
140156541479904 -> 140156541479792
140156541479904 [label=MkldnnConvolutionBackward]
140156541480016 -> 140156541479904
140156541480016 [label=ReluBackward1]
140156541480184 -> 140156541480016
140156541480184 [label=AddBackward0]
140156541480296 -> 140156541480184
140156541480296 [label=MulBackward0]
140156541480408 -> 140156541480296
140156541480408 [label=MkldnnConvolutionBackward]
140156541479176 -> 140156541480408
140156541479176 [label=ReluBackward1]
140156541480632 -> 140156541479176
140156541480632 [label=AddBackward0]
140156541480744 -> 140156541480632
140156541480744 [label=AddBackward0]
140156541480912 -> 140156541480744
140156541480912 [label=MulBackward0]
140156541501568 -> 140156541480912
140156541501568 [label=MkldnnConvolutionBackward]
140156541501680 -> 140156541501568
140156541501680 [label=ReluBackward1]
140156541501848 -> 140156541501680
140156541501848 [label=AddBackward0]
140156541501960 -> 140156541501848
140156541501960 [label=MulBackward0]
140156541502072 -> 140156541501960
140156541502072 [label=MkldnnConvolutionBackward]
140156541502184 -> 140156541502072
140156541502184 [label=ReluBackward1]
140156541502352 -> 140156541502184
140156541502352 [label=AddBackward0]
140156541502464 -> 140156541502352
140156541502464 [label=MulBackward0]
140156541502576 -> 140156541502464
140156541502576 [label=MkldnnConvolutionBackward]
140156541480800 -> 140156541502576
140156541480800 [label=ReluBackward1]
140156541502800 -> 140156541480800
140156541502800 [label=AddBackward0]
140156541502912 -> 140156541502800
140156541502912 [label=AddBackward0]
140156541503080 -> 140156541502912
140156541503080 [label=MulBackward0]
140156541503192 -> 140156541503080
140156541503192 [label=MkldnnConvolutionBackward]
140156541503304 -> 140156541503192
140156541503304 [label=ReluBackward1]
140156541503472 -> 140156541503304
140156541503472 [label=AddBackward0]
140156541503584 -> 140156541503472
140156541503584 [label=MulBackward0]
140156541503696 -> 140156541503584
140156541503696 [label=MkldnnConvolutionBackward]
140156541503808 -> 140156541503696
140156541503808 [label=ReluBackward1]
140156541503976 -> 140156541503808
140156541503976 [label=AddBackward0]
140156541504088 -> 140156541503976
140156541504088 [label=MulBackward0]
140156541504200 -> 140156541504088
140156541504200 [label=MkldnnConvolutionBackward]
140156541502968 -> 140156541504200
140156541502968 [label=ReluBackward1]
140156541504424 -> 140156541502968
140156541504424 [label=AddBackward0]
140156541504536 -> 140156541504424
140156541504536 [label=AddBackward0]
140156541504704 -> 140156541504536
140156541504704 [label=MulBackward0]
140156541504816 -> 140156541504704
140156541504816 [label=MkldnnConvolutionBackward]
140156541504928 -> 140156541504816
140156541504928 [label=ReluBackward1]
140156541505096 -> 140156541504928
140156541505096 [label=AddBackward0]
140156541505208 -> 140156541505096
140156541505208 [label=MulBackward0]
140156541505320 -> 140156541505208
140156541505320 [label=MkldnnConvolutionBackward]
140156541505432 -> 140156541505320
140156541505432 [label=ReluBackward1]
140156541513856 -> 140156541505432
140156541513856 [label=AddBackward0]
140156541513968 -> 140156541513856
140156541513968 [label=MulBackward0]
140156541514080 -> 140156541513968
140156541514080 [label=MkldnnConvolutionBackward]
140156541452528 -> 140156541514080
140156541514192 -> 140156541514080
140156541514192 [label="backbone.body.layer3.0.conv1.weight
(256, 512, 1, 1)" fillcolor=lightblue]
140156541505488 -> 140156541505320
140156541505488 [label="backbone.body.layer3.0.conv2.weight
(256, 256, 3, 3)" fillcolor=lightblue]
140156541504984 -> 140156541504816
140156541504984 [label="backbone.body.layer3.0.conv3.weight
(1024, 256, 1, 1)" fillcolor=lightblue]
140156541504592 -> 140156541504424
140156541504592 [label=AddBackward0]
140156541504760 -> 140156541504592
140156541504760 [label=MulBackward0]
140156541505040 -> 140156541504760
140156541505040 [label=MkldnnConvolutionBackward]
140156541452528 -> 140156541505040
140156541505376 -> 140156541505040
140156541505376 [label="backbone.body.layer3.0.downsample.0.weight
(1024, 512, 1, 1)" fillcolor=lightblue]
140156541504312 -> 140156541504200
140156541504312 [label="backbone.body.layer3.1.conv1.weight
(256, 1024, 1, 1)" fillcolor=lightblue]
140156541503864 -> 140156541503696
140156541503864 [label="backbone.body.layer3.1.conv2.weight
(256, 256, 3, 3)" fillcolor=lightblue]
140156541503360 -> 140156541503192
140156541503360 [label="backbone.body.layer3.1.conv3.weight
(1024, 256, 1, 1)" fillcolor=lightblue]
140156541502968 -> 140156541502800
140156541502688 -> 140156541502576
140156541502688 [label="backbone.body.layer3.2.conv1.weight
(256, 1024, 1, 1)" fillcolor=lightblue]
140156541502240 -> 140156541502072
140156541502240 [label="backbone.body.layer3.2.conv2.weight
(256, 256, 3, 3)" fillcolor=lightblue]
140156541501736 -> 140156541501568
140156541501736 [label="backbone.body.layer3.2.conv3.weight
(1024, 256, 1, 1)" fillcolor=lightblue]
140156541480800 -> 140156541480632
140156541480520 -> 140156541480408
140156541480520 [label="backbone.body.layer3.3.conv1.weight
(256, 1024, 1, 1)" fillcolor=lightblue]
140156541480072 -> 140156541479904
140156541480072 [label="backbone.body.layer3.3.conv2.weight
(256, 256, 3, 3)" fillcolor=lightblue]
140156541479568 -> 140156541479400
140156541479568 [label="backbone.body.layer3.3.conv3.weight
(1024, 256, 1, 1)" fillcolor=lightblue]
140156541479176 -> 140156541479008
140156541478896 -> 140156541478784
140156541478896 [label="backbone.body.layer3.4.conv1.weight
(256, 1024, 1, 1)" fillcolor=lightblue]
140156541478448 -> 140156541478280
140156541478448 [label="backbone.body.layer3.4.conv2.weight
(256, 256, 3, 3)" fillcolor=lightblue]
140156541477664 -> 140156541477776
140156541477664 [label="backbone.body.layer3.4.conv3.weight
(1024, 256, 1, 1)" fillcolor=lightblue]
140156541464528 -> 140156541463688
140156541463800 -> 140156541463184
140156541463800 [label="backbone.body.layer3.5.conv1.weight
(256, 1024, 1, 1)" fillcolor=lightblue]
140156541462792 -> 140156541462064
140156541462792 [label="backbone.body.layer3.5.conv2.weight
(256, 256, 3, 3)" fillcolor=lightblue]
140156541461672 -> 140156541460720
140156541461672 [label="backbone.body.layer3.5.conv3.weight
(1024, 256, 1, 1)" fillcolor=lightblue]
140156541455664 -> 140156541455328
140156541454656 -> 140156541453648
140156541454656 [label="backbone.fpn.fpn_inner3.weight
(256, 1024, 1, 1)" fillcolor=lightblue]
140156541454824 -> 140156541453648
140156541454824 [label="backbone.fpn.fpn_inner3.bias
(256)" fillcolor=lightblue]
140156541453088 -> 140156541452864
140156541453088 [label=UpsampleNearest2DBackward]
140156541454432 -> 140156541453088
140156541454432 [label=MkldnnConvolutionBackward]
140156541455888 -> 140156541454432
140156541455888 [label=ReluBackward1]
140156541462288 -> 140156541455888
140156541462288 [label=AddBackward0]
140156541461952 -> 140156541462288
140156541461952 [label=AddBackward0]
140156541463576 -> 140156541461952
140156541463576 [label=MulBackward0]
140156541463912 -> 140156541463576
140156541463912 [label=MkldnnConvolutionBackward]
140156541464024 -> 140156541463912
140156541464024 [label=ReluBackward1]
140156541478000 -> 140156541464024
140156541478000 [label=AddBackward0]
140156541478336 -> 140156541478000
140156541478336 [label=MulBackward0]
140156541478504 -> 140156541478336
140156541478504 [label=MkldnnConvolutionBackward]
140156541478840 -> 140156541478504
140156541478840 [label=ReluBackward1]
140156541479232 -> 140156541478840
140156541479232 [label=AddBackward0]
140156541479064 -> 140156541479232
140156541479064 [label=MulBackward0]
140156541479624 -> 140156541479064
140156541479624 [label=MkldnnConvolutionBackward]
140156541462512 -> 140156541479624
140156541462512 [label=ReluBackward1]
140156541480128 -> 140156541462512
140156541480128 [label=AddBackward0]
140156541480464 -> 140156541480128
140156541480464 [label=AddBackward0]
140156541480856 -> 140156541480464
140156541480856 [label=MulBackward0]
140156541501512 -> 140156541480856
140156541501512 [label=MkldnnConvolutionBackward]
140156541501792 -> 140156541501512
140156541501792 [label=ReluBackward1]
140156541501904 -> 140156541501792
140156541501904 [label=AddBackward0]
140156541502520 -> 140156541501904
140156541502520 [label=MulBackward0]
140156541502408 -> 140156541502520
140156541502408 [label=MkldnnConvolutionBackward]
140156541503024 -> 140156541502408
140156541503024 [label=ReluBackward1]
140156541503248 -> 140156541503024
140156541503248 [label=AddBackward0]
140156541503640 -> 140156541503248
140156541503640 [label=MulBackward0]
140156541503528 -> 140156541503640
140156541503528 [label=MkldnnConvolutionBackward]
140156541480240 -> 140156541503528
140156541480240 [label=ReluBackward1]
140156541504032 -> 140156541480240
140156541504032 [label=AddBackward0]
140156541504648 -> 140156541504032
140156541504648 [label=AddBackward0]
140156541505264 -> 140156541504648
140156541505264 [label=MulBackward0]
140156541514024 -> 140156541505264
140156541514024 [label=MkldnnConvolutionBackward]
140156541513800 -> 140156541514024
140156541513800 [label=ReluBackward1]
140156541514360 -> 140156541513800
140156541514360 [label=AddBackward0]
140156541514304 -> 140156541514360
140156541514304 [label=MulBackward0]
140156541514528 -> 140156541514304
140156541514528 [label=MkldnnConvolutionBackward]
140156541514640 -> 140156541514528
140156541514640 [label=ReluBackward1]
140156541514808 -> 140156541514640
140156541514808 [label=AddBackward0]
140156541514920 -> 140156541514808
140156541514920 [label=MulBackward0]
140156541515032 -> 140156541514920
140156541515032 [label=MkldnnConvolutionBackward]
140156541454208 -> 140156541515032
140156541515144 -> 140156541515032
140156541515144 [label="backbone.body.layer4.0.conv1.weight
(512, 1024, 1, 1)" fillcolor=lightblue]
140156541514696 -> 140156541514528
140156541514696 [label="backbone.body.layer4.0.conv2.weight
(512, 512, 3, 3)" fillcolor=lightblue]
140156541513912 -> 140156541514024
140156541513912 [label="backbone.body.layer4.0.conv3.weight
(2048, 512, 1, 1)" fillcolor=lightblue]
140156541504872 -> 140156541504032
140156541504872 [label=AddBackward0]
140156541505152 -> 140156541504872
140156541505152 [label=MulBackward0]
140156541514248 -> 140156541505152
140156541514248 [label=MkldnnConvolutionBackward]
140156541454208 -> 140156541514248
140156541514584 -> 140156541514248
140156541514584 [label="backbone.body.layer4.0.downsample.0.weight
(2048, 1024, 1, 1)" fillcolor=lightblue]
140156541504144 -> 140156541503528
140156541504144 [label="backbone.body.layer4.1.conv1.weight
(512, 2048, 1, 1)" fillcolor=lightblue]
140156541503136 -> 140156541502408
140156541503136 [label="backbone.body.layer4.1.conv2.weight
(512, 512, 3, 3)" fillcolor=lightblue]
140156541502016 -> 140156541501512
140156541502016 [label="backbone.body.layer4.1.conv3.weight
(2048, 512, 1, 1)" fillcolor=lightblue]
140156541480240 -> 140156541480128
140156541479960 -> 140156541479624
140156541479960 [label="backbone.body.layer4.2.conv1.weight
(512, 2048, 1, 1)" fillcolor=lightblue]
140156541478616 -> 140156541478504
140156541478616 [label="backbone.body.layer4.2.conv2.weight
(512, 512, 3, 3)" fillcolor=lightblue]
140156541477160 -> 140156541463912
140156541477160 [label="backbone.body.layer4.2.conv3.weight
(2048, 512, 1, 1)" fillcolor=lightblue]
140156541462512 -> 140156541462288
140156541461224 -> 140156541454432
140156541461224 [label="backbone.fpn.fpn_inner4.weight
(256, 2048, 1, 1)" fillcolor=lightblue]
140156541461280 -> 140156541454432
140156541461280 [label="backbone.fpn.fpn_inner4.bias
(256)" fillcolor=lightblue]
140156541357136 -> 140156541356968
140156541357136 [label="backbone.fpn.fpn_layer1.weight
(256, 256, 3, 3)" fillcolor=lightblue]
140156541357192 -> 140156541356968
140156541357192 [label="backbone.fpn.fpn_layer1.bias
(256)" fillcolor=lightblue]
140156541384600 -> 93906381257640
140156541384600 [label="(400, 5)" fillcolor=orange]
93907297852728 -> 140156541356688
93907297852728 [label=_ROIAlignBackward]
140156541356912 -> 93907297852728
140156541356912 [label=MkldnnConvolutionBackward]
140156541357640 -> 140156541356912
140156541357920 -> 140156541356912
140156541357920 [label="backbone.fpn.fpn_layer2.weight
(256, 256, 3, 3)" fillcolor=lightblue]
140156541357248 -> 140156541356912
140156541357248 [label="backbone.fpn.fpn_layer2.bias
(256)" fillcolor=lightblue]
140156541382944 -> 93907297852728
140156541382944 [label="(105, 5)" fillcolor=orange]
93906381121656 -> 140156541356576
93906381121656 [label=_ROIAlignBackward]
140156541356856 -> 93906381121656
140156541356856 [label=MkldnnConvolutionBackward]
140156541452864 -> 140156541356856
140156541357416 -> 140156541356856
140156541357416 [label="backbone.fpn.fpn_layer3.weight
(256, 256, 3, 3)" fillcolor=lightblue]
140156541453368 -> 140156541356856
140156541453368 [label="backbone.fpn.fpn_layer3.bias
(256)" fillcolor=lightblue]
140156541384024 -> 93906381121656
140156541384024 [label="(5, 5)" fillcolor=orange]
93907178273832 -> 140156541356464
93907178273832 [label=_ROIAlignBackward]
140156541356744 -> 93907178273832
140156541356744 [label=MkldnnConvolutionBackward]
140156541454432 -> 140156541356744
140156541456336 -> 140156541356744
140156541456336 [label="backbone.fpn.fpn_layer4.weight
(256, 256, 3, 3)" fillcolor=lightblue]
140156541453872 -> 140156541356744
140156541453872 [label="backbone.fpn.fpn_layer4.bias
(256)" fillcolor=lightblue]
140156541538376 -> 93907178273832
140156541538376 [label="(2, 5)" fillcolor=orange]
140156541356352 -> 140156541356016
140156541356352 [label=TBackward]
140156541356520 -> 140156541356352
140156541356520 [label="roi_heads.box.feature_extractor.fc6.weight
(1024, 12544)" fillcolor=lightblue]
140156541356128 -> 140156541355792
140156541356128 [label=TBackward]
140156541452416 -> 140156541356128
140156541452416 [label="roi_heads.box.feature_extractor.fc7.weight
(1024, 1024)" fillcolor=lightblue]
140156541355624 -> 140156541355904
140156541355624 [label=TBackward]
140156541355344 -> 140156541355624
140156541355344 [label="roi_heads.box.predictor.cls_score.weight
(81, 1024)" fillcolor=lightblue]
140156541354672 [label=DivBackward0 fillcolor=darkolivegreen1]
140156541355848 -> 140156541354672
140156541355848 [label=SumBackward0]
140156541355736 -> 140156541355848
140156541355736 [label=SWhereBackward]
140156541356632 -> 140156541355736
140156541356632 [label=DivBackward0]
140156541356408 -> 140156541356632
140156541356408 [label=MulBackward0]
140156541463072 -> 140156541356408
140156541463072 [label=PowBackward0]
140156541462400 -> 140156541463072
140156541462400 [label=AbsBackward]
140156541477888 -> 140156541462400
140156541477888 [label=SubBackward0]
140156541478728 -> 140156541477888
140156541478728 [label=IndexBackward]
140156541478952 -> 140156541478728
140156541478952 [label=AddmmBackward]
140156541479848 -> 140156541478952
140156541479848 [label="roi_heads.box.predictor.bbox_pred.bias
(324)" fillcolor=lightblue]
140156541355680 -> 140156541478952
140156541479344 -> 140156541478952
140156541479344 [label=TBackward]
140156541480576 -> 140156541479344
140156541480576 [label="roi_heads.box.predictor.bbox_pred.weight
(324, 1024)" fillcolor=lightblue]
140156541357024 -> 140156541355736
140156541357024 [label=SubBackward0]
140156541462400 -> 140156541357024
140156541355568 [label=BinaryCrossEntropyWithLogitsBackward fillcolor=darkolivegreen1]
140157898887520 -> 140156541355568
140157898887520 [label=IndexBackward]
140156541356240 -> 140157898887520
140156541356240 [label=MkldnnConvolutionBackward]
140157916263200 -> 140156541356240
140157916263200 [label=ReluBackward0]
140156541478168 -> 140157916263200
140156541478168 [label=ThnnConvTranspose2DBackward]
140156541479456 -> 140156541478168
140156541479456 [label=ReluBackward0]
140156541501624 -> 140156541479456
140156541501624 [label=MkldnnConvolutionBackward]
140156541502632 -> 140156541501624
140156541502632 [label=ReluBackward0]
140156541503752 -> 140156541502632
140156541503752 [label=MkldnnConvolutionBackward]
140156541503416 -> 140156541503752
140156541503416 [label=ReluBackward0]
140156541514136 -> 140156541503416
140156541514136 [label=MkldnnConvolutionBackward]
140156541514416 -> 140156541514136
140156541514416 [label=ReluBackward0]
140156541514864 -> 140156541514416
140156541514864 [label=MkldnnConvolutionBackward]
140156541515312 -> 140156541514864
140156541515312 [label=IndexPutBackward]
140156541515480 -> 140156541515312
140156541515480 [label=IndexPutBackward]
140156541515592 -> 140156541515480
140156541515592 [label=IndexPutBackward]
140156541515704 -> 140156541515592
140156541515704 [label=IndexPutBackward]
93907179961624 -> 140156541515704
93907179961624 [label=_ROIAlignBackward]
140156541356968 -> 93907179961624
140156541538448 -> 93907179961624
140156541538448 [label="(0, 5)" fillcolor=orange]
140156541515760 -> 140156541515592
140156541515760 [label=ViewBackward]
93907179960056 -> 140156541515760
93907179960056 [label=_ROIAlignBackward]
140156541356912 -> 93907179960056
140156541538592 -> 93907179960056
140156541538592 [label="(1, 5)" fillcolor=orange]
93906381217880 -> 140156541515480
93906381217880 [label=_ROIAlignBackward]
140156541356856 -> 93906381217880
140156541538520 -> 93906381217880
140156541538520 [label="(2, 5)" fillcolor=orange]
93906381223560 -> 140156541515312
93906381223560 [label=_ROIAlignBackward]
140156541356744 -> 93906381223560
140156541538664 -> 93906381223560
140156541538664 [label="(2, 5)" fillcolor=orange]
140156541515368 -> 140156541514864
140156541515368 [label="roi_heads.mask.feature_extractor.mask_fcn1.weight
(256, 256, 3, 3)" fillcolor=lightblue]
140156541515256 -> 140156541514864
140156541515256 [label="roi_heads.mask.feature_extractor.mask_fcn1.bias
(256)" fillcolor=lightblue]
140156541514976 -> 140156541514136
140156541514976 [label="roi_heads.mask.feature_extractor.mask_fcn2.weight
(256, 256, 3, 3)" fillcolor=lightblue]
140156541515088 -> 140156541514136
140156541515088 [label="roi_heads.mask.feature_extractor.mask_fcn2.bias
(256)" fillcolor=lightblue]
140156541504256 -> 140156541503752
140156541504256 [label="roi_heads.mask.feature_extractor.mask_fcn3.weight
(256, 256, 3, 3)" fillcolor=lightblue]
140156541504480 -> 140156541503752
140156541504480 [label="roi_heads.mask.feature_extractor.mask_fcn3.bias
(256)" fillcolor=lightblue]
140156541502744 -> 140156541501624
140156541502744 [label="roi_heads.mask.feature_extractor.mask_fcn4.weight
(256, 256, 3, 3)" fillcolor=lightblue]
140156541502296 -> 140156541501624
140156541502296 [label="roi_heads.mask.feature_extractor.mask_fcn4.bias
(256)" fillcolor=lightblue]
140156541479736 -> 140156541478168
140156541479736 [label="roi_heads.mask.predictor.conv5_mask.weight
(256, 256, 2, 2)" fillcolor=lightblue]
140156541480688 -> 140156541478168
140156541480688 [label="roi_heads.mask.predictor.conv5_mask.bias
(256)" fillcolor=lightblue]
140156541463408 -> 140156541356240
140156541463408 [label="roi_heads.mask.predictor.mask_fcn_logits.weight
(81, 256, 1, 1)" fillcolor=lightblue]
140156541461784 -> 140156541356240
140156541461784 [label="roi_heads.mask.predictor.mask_fcn_logits.bias
(81)" fillcolor=lightblue]
140156541355176 [label=BinaryCrossEntropyWithLogitsBackward fillcolor=darkolivegreen1]
140156541355960 -> 140156541355176
140156541355960 [label=IndexBackward]
140156541464136 -> 140156541355960
140156541464136 [label=AsStridedBackward]
140156541480352 -> 140156541464136
140156541480352 [label=CatBackward]
140156541502856 -> 140156541480352
140156541502856 [label=UnsafeViewBackward]
140156541514472 -> 140156541502856
140156541514472 [label=CloneBackward]
140156541515200 -> 140156541514472
140156541515200 [label=PermuteBackward]
140156541515536 -> 140156541515200
140156541515536 [label=MkldnnConvolutionBackward]
140156541515872 -> 140156541515536
140156541515872 [label=ReluBackward0]
140156541516152 -> 140156541515872
140156541516152 [label=MkldnnConvolutionBackward]
140156541356968 -> 140156541516152
140156541516264 -> 140156541516152
140156541516264 [label="rpn.head.conv.weight
(256, 256, 3, 3)" fillcolor=lightblue]
140156541516320 -> 140156541516152
140156541516320 [label="rpn.head.conv.bias
(256)" fillcolor=lightblue]
140156541515984 -> 140156541515536
140156541515984 [label="rpn.head.cls_logits.weight
(3, 256, 1, 1)" fillcolor=lightblue]
140156541516040 -> 140156541515536
140156541516040 [label="rpn.head.cls_logits.bias
(3)" fillcolor=lightblue]
140156541504368 -> 140156541480352
140156541504368 [label=UnsafeViewBackward]
140156541515648 -> 140156541504368
140156541515648 [label=CloneBackward]
140156541515816 -> 140156541515648
140156541515816 [label=PermuteBackward]
140156541516376 -> 140156541515816
140156541516376 [label=MkldnnConvolutionBackward]
140156541516488 -> 140156541516376
140156541516488 [label=ReluBackward0]
140156541516432 -> 140156541516488
140156541516432 [label=MkldnnConvolutionBackward]
140156541356912 -> 140156541516432
140156541516264 -> 140156541516432
140156541516320 -> 140156541516432
140156541515984 -> 140156541516376
140156541516040 -> 140156541516376
140156541502128 -> 140156541480352
140156541502128 [label=UnsafeViewBackward]
140156541515928 -> 140156541502128
140156541515928 [label=CloneBackward]
140156541516208 -> 140156541515928
140156541516208 [label=PermuteBackward]
140156541516600 -> 140156541516208
140156541516600 [label=MkldnnConvolutionBackward]
140156541516712 -> 140156541516600
140156541516712 [label=ReluBackward0]
140156541516824 -> 140156541516712
140156541516824 [label=MkldnnConvolutionBackward]
140156541356856 -> 140156541516824
140156541516264 -> 140156541516824
140156541516320 -> 140156541516824
140156541515984 -> 140156541516600
140156541516040 -> 140156541516600
140156541503920 -> 140156541480352
140156541503920 [label=UnsafeViewBackward]
140156541516096 -> 140156541503920
140156541516096 [label=CloneBackward]
140156541516656 -> 140156541516096
140156541516656 [label=PermuteBackward]
140156541516880 -> 140156541516656
140156541516880 [label=MkldnnConvolutionBackward]
140156541516992 -> 140156541516880
140156541516992 [label=ReluBackward0]
140156541517104 -> 140156541516992
140156541517104 [label=MkldnnConvolutionBackward]
140156541356744 -> 140156541517104
140156541516264 -> 140156541517104
140156541516320 -> 140156541517104
140156541515984 -> 140156541516880
140156541516040 -> 140156541516880
140156541514752 -> 140156541480352
140156541514752 [label=UnsafeViewBackward]
140156541516544 -> 140156541514752
140156541516544 [label=CloneBackward]
140156541516936 -> 140156541516544
140156541516936 [label=PermuteBackward]
140156541517160 -> 140156541516936
140156541517160 [label=MkldnnConvolutionBackward]
140156541517272 -> 140156541517160
140156541517272 [label=ReluBackward0]
140156541517384 -> 140156541517272
140156541517384 [label=MkldnnConvolutionBackward]
140156541517496 -> 140156541517384
140156541517496 [label=MaxPool2DWithIndicesBackward]
140156541356744 -> 140156541517496
140156541516264 -> 140156541517384
140156541516320 -> 140156541517384
140156541515984 -> 140156541517160
140156541516040 -> 140156541517160
140156541355400 [label=DivBackward0 fillcolor=darkolivegreen1]
140156541356072 -> 140156541355400
140156541356072 [label=SumBackward0]
140157898858392 -> 140156541356072
140157898858392 [label=SWhereBackward]
140156541516768 -> 140157898858392
140156541516768 [label=DivBackward0]
140156541517328 -> 140156541516768
140156541517328 [label=MulBackward0]
140156541517552 -> 140156541517328
140156541517552 [label=PowBackward0]
140156541517664 -> 140156541517552
140156541517664 [label=AbsBackward]
140156541517776 -> 140156541517664
140156541517776 [label=SubBackward0]
140156541563008 -> 140156541517776
140156541563008 [label=IndexBackward]
140156541563120 -> 140156541563008
140156541563120 [label=AsStridedBackward]
140156541563232 -> 140156541563120
140156541563232 [label=CatBackward]
140156541563344 -> 140156541563232
140156541563344 [label=UnsafeViewBackward]
140156541563680 -> 140156541563344
140156541563680 [label=CloneBackward]
140156541563792 -> 140156541563680
140156541563792 [label=PermuteBackward]
140156541563904 -> 140156541563792
140156541563904 [label=ViewBackward]
140156541564016 -> 140156541563904
140156541564016 [label=MkldnnConvolutionBackward]
140156541515872 -> 140156541564016
140156541564128 -> 140156541564016
140156541564128 [label="rpn.head.bbox_pred.weight
(12, 256, 1, 1)" fillcolor=lightblue]
140156541564184 -> 140156541564016
140156541564184 [label="rpn.head.bbox_pred.bias
(12)" fillcolor=lightblue]
140156541563400 -> 140156541563232
140156541563400 [label=UnsafeViewBackward]
140156541563736 -> 140156541563400
140156541563736 [label=CloneBackward]
140156541563960 -> 140156541563736
140156541563960 [label=PermuteBackward]
140156541564240 -> 140156541563960
140156541564240 [label=ViewBackward]
140156541564408 -> 140156541564240
140156541564408 [label=MkldnnConvolutionBackward]
140156541516488 -> 140156541564408
140156541564128 -> 140156541564408
140156541564184 -> 140156541564408
140156541563456 -> 140156541563232
140156541563456 [label=UnsafeViewBackward]
140156541563848 -> 140156541563456
140156541563848 [label=CloneBackward]
140156541564352 -> 140156541563848
140156541564352 [label=PermuteBackward]
140156541564464 -> 140156541564352
140156541564464 [label=ViewBackward]
140156541564576 -> 140156541564464
140156541564576 [label=MkldnnConvolutionBackward]
140156541516712 -> 140156541564576
140156541564128 -> 140156541564576
140156541564184 -> 140156541564576
140156541563512 -> 140156541563232
140156541563512 [label=UnsafeViewBackward]
140156541564072 -> 140156541563512
140156541564072 [label=CloneBackward]
140156541564520 -> 140156541564072
140156541564520 [label=PermuteBackward]
140156541564688 -> 140156541564520
140156541564688 [label=ViewBackward]
140156541564800 -> 140156541564688
140156541564800 [label=MkldnnConvolutionBackward]
140156541516992 -> 140156541564800
140156541564128 -> 140156541564800
140156541564184 -> 140156541564800
140156541563568 -> 140156541563232
140156541563568 [label=UnsafeViewBackward]
140156541564296 -> 140156541563568
140156541564296 [label=CloneBackward]
140156541564744 -> 140156541564296
140156541564744 [label=PermuteBackward]
140156541564912 -> 140156541564744
140156541564912 [label=ViewBackward]
140156541565024 -> 140156541564912
140156541565024 [label=MkldnnConvolutionBackward]
140156541517272 -> 140156541565024
140156541564128 -> 140156541565024
140156541564184 -> 140156541565024
140156541517048 -> 140157898858392
140156541517048 [label=SubBackward0]
140156541517664 -> 140156541517048
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment