On the setting of batch size in test_loader

[Lyman-Smoker]

Hi, Thanks for your great work!

I try to run your the released code on FineDiving dataset, but when I try to adjust the batch size of test_loader from 1 to any other number (e.g., 3, 7 and 8), it will post the following error:

I am wondering if there is some way to fix this problem since the inference process is too slow when the batch size of test_loader is set to 1. Could you please offer some hints for this problem?

Thank you in advance!