Skip to content

Commit aa3b00c

Browse files
committed
add TODO for issues 288
1 parent cd6791e commit aa3b00c

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tensorlayer/utils.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -300,7 +300,7 @@ def predict(sess, network, X, x, y_op, batch_size=None):
300300
if result is None:
301301
result = result_a
302302
else:
303-
result = np.vstack((result, result_a))
303+
result = np.vstack((result, result_a)) # TODO: https://github.com/tensorlayer/tensorlayer/issues/288
304304
if result is None:
305305
if len(X) % batch_size != 0:
306306
dp_dict = dict_to_one(network.all_drop)
@@ -318,7 +318,7 @@ def predict(sess, network, X, x, y_op, batch_size=None):
318318
}
319319
feed_dict.update(dp_dict)
320320
result_a = sess.run(y_op, feed_dict=feed_dict)
321-
result = np.vstack((result, result_a))
321+
result = np.vstack((result, result_a)) # TODO: https://github.com/tensorlayer/tensorlayer/issues/288
322322
return result
323323

324324

0 commit comments

Comments
 (0)