atari-rl-algos/rnn_torchviz at main · pooleycodes/atari-rl-algos · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
digraph {
	graph [size="12,12"]
	node [align=left fontname=monospace fontsize=10 height=0.2 ranksep=0.1 shape=box style=filled]
	139662073928096 [label="
 (1, 6)" fillcolor=darkolivegreen1]
	139663908303008 [label=SoftmaxBackward0]
	139663908303440 -> 139663908303008
	139663908303440 [label=AddmmBackward0]
	139663908195584 -> 139663908303440
	139662073805696 [label="actor.2.bias
 (6)" fillcolor=lightblue]
	139662073805696 -> 139663908195584
	139663908195584 [label=AccumulateGrad]
	139663908304400 -> 139663908303440
	139663908304400 [label=ReluBackward0]
	139663908303776 -> 139663908304400
	139663908303776 [label=AddmmBackward0]
	139663908193280 -> 139663908303776
	139662073805536 [label="actor.0.bias
 (512)" fillcolor=lightblue]
	139662073805536 -> 139663908193280
	139663908193280 [label=AccumulateGrad]
	139663908304256 -> 139663908303776
	139663908304256 [label=ViewBackward0]
	139663908304352 -> 139663908304256
	139663908304352 [label=ReluBackward0]
	139663908304592 -> 139663908304352
	139663908304592 [label=AddBackward0]
	139663908304688 -> 139663908304592
	139663908304688 [label=CudnnConvolutionBackward0]
	139663908304832 -> 139663908304688
	139663908304832 [label=ReluBackward0]
	139663908304928 -> 139663908304832
	139663908304928 [label=AddBackward0]
	139663908305024 -> 139663908304928
	139663908305024 [label=CudnnConvolutionBackward0]
	139663908305168 -> 139663908305024
	139663908305168 [label=ReluBackward0]
	139663908305264 -> 139663908305168
	139663908305264 [label=AddBackward0]
	139663908305360 -> 139663908305264
	139663908305360 [label=CudnnConvolutionBackward0]
	139663908192656 -> 139663908305360
	139662073804016 [label="conv.0.weight
 (32, 4, 8, 8)" fillcolor=lightblue]
	139662073804016 -> 139663908192656
	139663908192656 [label=AccumulateGrad]
	139663908305312 -> 139663908305264
	139663908305312 [label=ReshapeAliasBackward0]
	139663908195824 -> 139663908305312
	139662073804496 [label="conv.0.bias
 (32)" fillcolor=lightblue]
	139662073804496 -> 139663908195824
	139663908195824 [label=AccumulateGrad]
	139663908195440 -> 139663908305024
	139662073804656 [label="conv.2.weight
 (64, 32, 4, 4)" fillcolor=lightblue]
	139662073804656 -> 139663908195440
	139663908195440 [label=AccumulateGrad]
	139663908304976 -> 139663908304928
	139663908304976 [label=ReshapeAliasBackward0]
	139663908193136 -> 139663908304976
	139662073804736 [label="conv.2.bias
 (64)" fillcolor=lightblue]
	139662073804736 -> 139663908193136
	139663908193136 [label=AccumulateGrad]
	139663908195968 -> 139663908304688
	139662073804896 [label="conv.4.weight
 (64, 64, 3, 3)" fillcolor=lightblue]
	139662073804896 -> 139663908195968
	139663908195968 [label=AccumulateGrad]
	139663908304640 -> 139663908304592
	139663908304640 [label=ReshapeAliasBackward0]
	139663908195872 -> 139663908304640
	139662073804976 [label="conv.4.bias
 (64)" fillcolor=lightblue]
	139662073804976 -> 139663908195872
	139663908195872 [label=AccumulateGrad]
	139663908303680 -> 139663908303776
	139663908303680 [label=TBackward0]
	139663908192512 -> 139663908303680
	139662073805136 [label="actor.0.weight
 (512, 3136)" fillcolor=lightblue]
	139662073805136 -> 139663908192512
	139663908192512 [label=AccumulateGrad]
	139663908304304 -> 139663908303440
	139663908304304 [label=TBackward0]
	139663908194528 -> 139663908304304
	139662073805616 [label="actor.2.weight
 (6, 512)" fillcolor=lightblue]
	139662073805616 -> 139663908194528
	139663908194528 [label=AccumulateGrad]
	139663908303008 -> 139662073928096
}