We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 903ebcc commit 1553f3aCopy full SHA for 1553f3a
eval.py
@@ -49,6 +49,7 @@ def evaluate(beam_size):
49
batch_size=1, shuffle=True, num_workers=1, pin_memory=True)
50
51
# TODO: Batched Beam Search
52
+ # Therefore, do not use a batch_size greater than 1 during evaluation - IMPORTANT
53
54
# Lists to store references (true captions), and hypothesis (prediction) for each image
55
# If for n images, we have n hypotheses, and references a, b, c... for each image, we need -
0 commit comments