-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathpolicy_loss
121 lines (121 loc) · 3.34 KB
/
policy_loss
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
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
digraph {
graph [size="15.45,15.45"]
node [align=left fontsize=12 height=0.2 ranksep=0.1 shape=box style=filled]
5376974928 [label=MeanBackward0 fillcolor=darkolivegreen1]
5376974992 -> 5376974928
5376974992 [label=SubBackward0]
5376975120 -> 5376974992
5376975120 [label=MulBackward0]
5376975312 -> 5376975120
5376975312 [label=SubBackward0]
5376975504 -> 5376975312
5376975504 [label=SubBackward0]
5376975632 -> 5376975504
5376975632 [label=SubBackward0]
5376975440 -> 5376975632
5376975440 [label=DivBackward0]
5365944144 -> 5376975440
5365944144 [label=NegBackward]
5365983248 -> 5365944144
5365983248 [label=PowBackward0]
5376976080 -> 5365983248
5376976080 [label=SubBackward0]
5376976208 -> 5376976080
5376976208 [label=AddmmBackward]
5376976336 -> 5376976208
5376976336 [label="
(1)" fillcolor=lightblue]
5376976400 -> 5376976208
5376976400 [label=ReluBackward0]
5376976656 -> 5376976400
5376976656 [label=AddmmBackward]
5376977040 -> 5376976656
5376977040 [label="
(256)" fillcolor=lightblue]
5376976976 -> 5376976656
5376976976 [label=ReluBackward0]
5376977232 -> 5376976976
5376977232 [label=AddmmBackward]
5376977360 -> 5376977232
5376977360 [label="
(256)" fillcolor=lightblue]
5376976720 -> 5376977232
5376976720 [label=TBackward]
5376977424 -> 5376976720
5376977424 [label="
(256, 3)" fillcolor=lightblue]
5376976912 -> 5376976656
5376976912 [label=TBackward]
5376976784 -> 5376976912
5376976784 [label="
(256, 256)" fillcolor=lightblue]
5376976464 -> 5376976208
5376976464 [label=TBackward]
5376976592 -> 5376976464
5376976592 [label="
(1, 256)" fillcolor=lightblue]
5376975888 -> 5376975440
5376975888 [label=MulBackward0]
5376976016 -> 5376975888
5376976016 [label=PowBackward0]
5376976272 -> 5376976016
5376976272 [label=ExpBackward]
5376976848 -> 5376976272
5376976848 [label=AddBackward0]
5376977104 -> 5376976848
5376977104 [label=MulBackward0]
5376977616 -> 5376977104
5376977616 [label=AddBackward0]
5376977680 -> 5376977616
5376977680 [label=TanhBackward]
5376977808 -> 5376977680
5376977808 [label=AddmmBackward]
5376977936 -> 5376977808
5376977936 [label="
(1)" fillcolor=lightblue]
5376976400 -> 5376977808
5376978000 -> 5376977808
5376978000 [label=TBackward]
5376978192 -> 5376978000
5376978192 [label="
(1, 256)" fillcolor=lightblue]
5376975760 -> 5376975632
5376975760 [label=LogBackward]
5376976272 -> 5376975760
5376975184 -> 5376974992
5376975184 [label=AddmmBackward]
5376975376 -> 5376975184
5376975376 [label="
(1)" fillcolor=lightblue]
5376975568 -> 5376975184
5376975568 [label=ReluBackward0]
5376976144 -> 5376975568
5376976144 [label=AddmmBackward]
5376977296 -> 5376976144
5376977296 [label="
(256)" fillcolor=lightblue]
5376978384 -> 5376976144
5376978384 [label=CatBackward]
5376977872 -> 5376978384
5376977872 [label=ReluBackward0]
5376978128 -> 5376977872
5376978128 [label=AddmmBackward]
5376978320 -> 5376978128
5376978320 [label="
(256)" fillcolor=lightblue]
5376977552 -> 5376978128
5376977552 [label=TBackward]
5376978512 -> 5376977552
5376978512 [label="
(256, 3)" fillcolor=lightblue]
5376977168 -> 5376976144
5376977168 [label=TBackward]
5376977744 -> 5376977168
5376977744 [label="
(256, 257)" fillcolor=lightblue]
5376975696 -> 5376975184
5376975696 [label=TBackward]
5376975952 -> 5376975696
5376975952 [label="
(1, 256)" fillcolor=lightblue]
}