Skip to content

Commit

Permalink
Fix typo
Browse files Browse the repository at this point in the history
  • Loading branch information
ysk24ok committed Feb 28, 2020
1 parent df8b4b5 commit b48bdbf
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions ch07/train_seq2seq.py
Original file line number Diff line number Diff line change
Expand Up @@ -24,14 +24,14 @@
# ハイパーパラメータの設定
vocab_size = len(char_to_id)
wordvec_size = 16
hideen_size = 128
hidden_size = 128
batch_size = 128
max_epoch = 25
max_grad = 5.0

# Normal or Peeky? ==============================================
model = Seq2seq(vocab_size, wordvec_size, hideen_size)
# model = PeekySeq2seq(vocab_size, wordvec_size, hideen_size)
model = Seq2seq(vocab_size, wordvec_size, hidden_size)
# model = PeekySeq2seq(vocab_size, wordvec_size, hidden_size)
# ================================================================
optimizer = Adam()
trainer = Trainer(model, optimizer)
Expand Down

0 comments on commit b48bdbf

Please sign in to comment.