Skip to content
Permalink
Browse files

Fix(batch_size): fix for loop index error on batch_size

  • Loading branch information...
Armour committed Nov 28, 2018
1 parent b5c0703 commit e4a75b7fcdee4b046169ad5aee0e773cd1f4cd78
Showing with 1 addition and 1 deletion.
  1. +1 −1 test.py
@@ -42,7 +42,7 @@
avg_loss += l

# Save all testing image
for i in range(batch_size):
for i in range(len(color)):
summary_image = concat_images(gray[i], pred[i])
summary_image = concat_images(summary_image, color[i])
plt.imsave("%s/images/%d_%d.png" % (testing_summary, step, i), summary_image)

0 comments on commit e4a75b7

Please sign in to comment.
You can’t perform that action at this time.