From e4d96ed044c56cf0ae9dd618b5182bfab0e0a991 Mon Sep 17 00:00:00 2001
From: tfavory <43109272+tfavory@users.noreply.github.com>
Date: Mon, 10 Sep 2018 23:39:12 +0800
Subject: [PATCH] Update poem_generator.py

---
 poem_generator.py | 11 ++---------
 1 file changed, 2 insertions(+), 9 deletions(-)

diff --git a/poem_generator.py b/poem_generator.py
index 128d09d..0545251 100644
--- a/poem_generator.py
+++ b/poem_generator.py
@@ -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):