Create utils.py

Author: Nandayang

tensorlayer/utils.py

@@ -130,15 +130,15 @@ def fit(sess, network, train_op, cost, X_train, y_train, x, y_, acc=None, batch_
                     result = sess.run(merged, feed_dict=feed_dict)
                     train_writer.add_summary(result, tensorboard_train_index)
                     tensorboard_train_index += 1
-
-                for X_val_a, y_val_a in iterate.minibatches(
+                if (X_val is not None) and (y_val is not None):                      
+                        for X_val_a, y_val_a in iterate.minibatches(
                                         X_val, y_val, batch_size, shuffle=True):
-                    dp_dict = dict_to_one( network.all_drop )    # disable noise layers
-                    feed_dict = {x: X_val_a, y_: y_val_a}
-                    feed_dict.update(dp_dict)
-                    result = sess.run(merged, feed_dict=feed_dict)
-                    val_writer.add_summary(result, tensorboard_val_index)
-                    tensorboard_val_index += 1
+                                dp_dict = dict_to_one( network.all_drop )    # disable noise layers
+                                feed_dict = {x: X_val_a, y_: y_val_a}
+                                feed_dict.update(dp_dict)
+                                result = sess.run(merged, feed_dict=feed_dict)
+                                val_writer.add_summary(result, tensorboard_val_index)
+                                tensorboard_val_index += 1
 
         if epoch + 1 == 1 or (epoch + 1) % print_freq == 0:
             if (X_val is not None) and (y_val is not None):