tf.train.SummaryWriter renamed

yenchenlin · Mar 20, 2017 · 971742b · 971742b
1 parent 80de637
commit 971742b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/model.py b/model.py
@@ -142,7 +142,7 @@ def train(self, args):
         self.g_sum = tf.summary.merge([self.d__sum,
             self.fake_B_sum, self.d_loss_fake_sum, self.g_loss_sum])
         self.d_sum = tf.summary.merge([self.d_sum, self.d_loss_real_sum, self.d_loss_sum])
-        self.writer = tf.train.SummaryWriter("./logs", self.sess.graph)
+        self.writer = tf.summary.FileWriter("./logs", self.sess.graph)
 
         counter = 1
         start_time = time.time()