diff --git a/ch07/train_seq2seq.py b/ch07/train_seq2seq.py index 167d703f..883dfd02 100644 --- a/ch07/train_seq2seq.py +++ b/ch07/train_seq2seq.py @@ -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)