PocketFlow icon indicating copy to clipboard operation
PocketFlow copied to clipboard

Published 20 hours ago • verified publisher icon Tencent

Change "batch_size" to "batch_size_eval" in __build_eval function

Open rapwag01 opened this issue 6 years ago • 1 comments

Describe the bug The __build_eval function in learner.py uses batch_size instead of batch_eval_size for the shape of evaluation images, ie

images.set_shape((FLAGS.batch_size, images.shape[1], images.shape[2],
needs to be changed to

images.set_shape((FLAGS.batch_size_eval, images.shape[1], images.shape[2], \

rapwag01 avatar Feb 20 '19 16:02 rapwag01

Thanks a lot. Bugs confirmed in these two files:

  • learners/uniform_quantization/learner.py
  • learners/nonuniform_quantization/learner.py

We will push a fix ASAP.

jiaxiang-wu avatar Feb 21 '19 00:02 jiaxiang-wu