device placement off

SamKirkiles · Apr 19, 2018 · 618c54e · 618c54e
1 parent 2ecf8dc
commit 618c54e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lstm.py b/lstm.py
@@ -128,7 +128,7 @@ def train(self,train_step,iterations=2000,restore=False):
 		with tf.device(self.light_device):
 			saver = tf.train.Saver()
 
-		with tf.Session(config=tf.ConfigProto(allow_soft_placement=True, log_device_placement=True)) as sess:
+		with tf.Session(config=tf.ConfigProto(allow_soft_placement=True)) as sess:
 			try:
 
 				if restore:
@@ -175,7 +175,7 @@ def generate(self,char2ix,ix2char,seq_length,restore=False):
 
 			saver = tf.train.Saver()
 
-		with tf.Session(config=tf.ConfigProto(allow_soft_placement=True, log_device_placement=True)) as sess:
+		with tf.Session(config=tf.ConfigProto(allow_soft_placement=True)) as sess:
 
 			# init session
 			if restore: