forked from curiousily/Getting-Things-Done-with-Pytorch
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathLSTM_model
50 lines (50 loc) · 1.67 KB
/
LSTM_model
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
digraph {
graph [size="12,12"]
node [align=left fontname=monospace fontsize=10 height=0.2 ranksep=0.1 shape=box style=filled]
2283802772080 [label="
(1, 1)" fillcolor=darkolivegreen1]
2283802592880 [label=AddmmBackward0]
2283802592688 -> 2283802592880
2283802772000 [label="fc.bias
(1)" fillcolor=lightblue]
2283802772000 -> 2283802592688
2283802592688 [label=AccumulateGrad]
2283802595280 -> 2283802592880
2283802595280 [label=SliceBackward0]
2283802593312 -> 2283802595280
2283802593312 [label=SelectBackward0]
2283802592112 -> 2283802593312
2283802592112 [label=SliceBackward0]
2283802593504 -> 2283802592112
2283802593504 [label=TransposeBackward0]
2283802592928 -> 2283802593504
2283802592928 [label=MkldnnRnnLayerBackward0]
2283802593408 -> 2283802592928
2283656326896 [label="lstm.weight_ih_l0
(200, 1)" fillcolor=lightblue]
2283656326896 -> 2283802593408
2283802593408 [label=AccumulateGrad]
2283802593744 -> 2283802592928
2283656326816 [label="lstm.weight_hh_l0
(200, 50)" fillcolor=lightblue]
2283656326816 -> 2283802593744
2283802593744 [label=AccumulateGrad]
2283802593648 -> 2283802592928
2283656326976 [label="lstm.bias_ih_l0
(200)" fillcolor=lightblue]
2283656326976 -> 2283802593648
2283802593648 [label=AccumulateGrad]
2283802594032 -> 2283802592928
2283656327056 [label="lstm.bias_hh_l0
(200)" fillcolor=lightblue]
2283656327056 -> 2283802594032
2283802594032 [label=AccumulateGrad]
2283802594992 -> 2283802592880
2283802594992 [label=TBackward0]
2283802593024 -> 2283802594992
2283802771920 [label="fc.weight
(1, 50)" fillcolor=lightblue]
2283802771920 -> 2283802593024
2283802593024 [label=AccumulateGrad]
2283802592880 -> 2283802772080
}