Skip to content

Commit 1553f3a

Browse files
committed
eval.py warning
1 parent 903ebcc commit 1553f3a

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

eval.py

+1
Original file line numberDiff line numberDiff line change
@@ -49,6 +49,7 @@ def evaluate(beam_size):
4949
batch_size=1, shuffle=True, num_workers=1, pin_memory=True)
5050

5151
# TODO: Batched Beam Search
52+
# Therefore, do not use a batch_size greater than 1 during evaluation - IMPORTANT
5253

5354
# Lists to store references (true captions), and hypothesis (prediction) for each image
5455
# If for n images, we have n hypotheses, and references a, b, c... for each image, we need -

0 commit comments

Comments
 (0)