wassname/erotic-generator
1
0
Fork 0
mirror of https://github.com/wassname/erotic-generator.git synced 2026-09-17 12:10:13 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update poem_generator.py

Browse Source
This commit is contained in:
tfavory
2018-09-10 23:39:12 +08:00
committed by GitHub
parent 4f811d632d
commit e4d96ed044
1 changed files with 2 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
poem_generator.py
+2 -9
View File
@@ -45,16 +45,9 @@ class Model(tf.keras.Model):
self.batch_sz = batch_size
self.embedding = tf.keras.layers.Embedding(vocab_size, embedding_dim)
if tf.test.is_gpu_available():
self.gru = tf.keras.layers.CuDNNGRU(self.units,
return_sequences=True,
return_state=True,
recurrent_initializer='glorot_uniform')
self.gru = tf.keras.layers.CuDNNGRU(self.units, return_sequences=True, return_state=True, recurrent_initializer='glorot_uniform')
else:
self.gru = tf.keras.layers.GRU(self.units,
return_sequences=True,
return_state=True,
recurrent_activation='sigmoid',
recurrent_initializer='glorot_uniform')
self.gru = tf.keras.layers.GRU(self.units, return_sequences=True, return_state=True, recurrent_activation='sigmoid', recurrent_initializer='glorot_uniform')
self.fc = tf.keras.layers.Dense(vocab_size)
def call(self, x, hidden):