Update tutorial_generate_text.py

Author: zsdonghao

examples/text_generation/tutorial_generate_text.py

@@ -247,7 +247,7 @@ def main_lstm_generate_text():
 
     net = Text_Generation_Net(vocab_size, hidden_size, init)
 
-    train_weights = net.weights
+    train_weights = net.trainable_weights
     optimizer = tf.optimizers.Adam(lr=learning_rate)
 
     # ===== Training
@@ -329,4 +329,4 @@ def main_lstm_generate_text():
     # main_restore_embedding_layer()
 
     # How to generate text from a given context
-    main_lstm_generate_text()
+    main_lstm_generate_text()