Skip to content

Commit 9036bfd

Browse files
committed
reduced size of a test layer LSTM
1 parent 3198c5b commit 9036bfd

File tree

6 files changed

+2
-2
lines changed

6 files changed

+2
-2
lines changed
-3.88 KB
Binary file not shown.
-592 Bytes
Binary file not shown.
-592 Bytes
Binary file not shown.
-1.16 KB
Binary file not shown.

testdata/dnn/onnx/generate_onnx_models.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1155,8 +1155,8 @@ def forward(self, x, h, c):
11551155
return self.lstm(x, (h, c))[0]
11561156

11571157
batch = 1
1158-
features = 512
1159-
hidden = 156
1158+
features = 16
1159+
hidden = 8
11601160
seq_len = 2
11611161
num_layers = 1
11621162
bidirectional = False
-1.59 MB
Binary file not shown.

0 commit comments

Comments
 (0)