Skip to content

Commit eb23ef1

Browse files
Fixed bug with models apart from LSTMs
1 parent 427eb4d commit eb23ef1

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

word_language_model/model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -37,4 +37,4 @@ def initHidden(self, bsz):
3737
return (Variable(weight.new(self.nlayers, bsz, self.nhid).zero_()),
3838
Variable(weight.new(self.nlayers, bsz, self.nhid).zero_()))
3939
else:
40-
return Variable(weight.new(args.nlayers, bsz, args.nhid).zero_())
40+
return Variable(weight.new(self.nlayers, bsz, self.nhid).zero_())

0 commit comments

Comments
 (0)