add seq2seq model; add seq2seq test

tensorlayer · May 25, 2019 · 6401d54 · 6401d54
1 parent 8989efc
commit 6401d54
Show file tree

Hide file tree

Showing 5 changed files with 10,033 additions and 20 deletions.
diff --git a/examples/text_generation/tutorial_generate_text.py b/examples/text_generation/tutorial_generate_text.py
@@ -266,9 +266,12 @@ def main_lstm_generate_text():
         # reset all states at the begining of every epoch
         lstm_state = None
         for step, (x, y) in enumerate(tl.iterate.ptb_iterator(train_data, batch_size, sequence_length)):
+            print(">>>>>", y)
             with tf.GradientTape() as tape:
+
                 ## compute outputs
                 logits, lstm_state = net(x, initial_state=lstm_state)
+                print(">>>>logits" , logits)
                 ## compute loss and update model
                 cost = tl.cost.cross_entropy(logits, tf.reshape(y, [-1]), name='train_loss')