alihan/TensorFlow-2.x-Tutorials
1
0
Fork 0
mirror of https://github.com/dragen1860/TensorFlow-2.x-Tutorials.git synced 2021-05-12 18:32:23 +03:00
Code Issues Projects Releases Wiki Activity

Update gru_sentiment_analysis_layer.py

Browse Source 
enable training for rnn layer
This commit is contained in:
kelvinkoh0308
2019-09-09 22:09:32 +09:00
committed by GitHub
parent dbd0334397
commit df1e550fa8
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
深度学习与TensorFlow入门实战-源码和PPT/lesson47-LSTM实战/gru_sentiment_analysis_layer.py
View File

@@ -73,7 +73,7 @@ class MyRNN(keras.Model):
x = self.embedding(x)
# rnn cell compute
# x: [b, 80, 100] => [b, 64]
x = self.rnn(x)
x = self.rnn(x,training=training)
# out: [b, 64] => [b, 1]
x = self.outlayer(x)