yashkant/spad

Assertion Error

Closed this issue · 2 comments

Tzy010822 commented

It seems solved when

spad/scripts/inference.py

Line 159 in 4c9aa18

if batch_size * (idx + 1) > len(captions):
changes to >=.

yashkant commented

fixed, thanks!