Skip to content

Commit a217fef

Browse files
bdfhjkapaszke
authored andcommitted
Update rnn.py
Fixed a problem with outputting the RuntimeError if arguments are incorrect in cudnn/rnn.py
1 parent 5221745 commit a217fef

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

torch/backends/cudnn/rnn.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -198,7 +198,7 @@ def forward(fn, input, hx, weight, output, hy):
198198
'input must have 3 dimensions, got {}'.format(input.dim()))
199199
if fn.input_size != input.size(2):
200200
raise RuntimeError('input.size(2) must be equal to input_size. Expected {}, got {}'.format(
201-
fn.input_size
201+
fn.input_size, input.size(2)
202202
))
203203
if fn.dropout != 0 and cudnn.version() < 5103:
204204
raise RuntimeError('dropout supported only in cudnn v5.1 and above')

0 commit comments

Comments
 (0)