xujinglin / FineDiving

FineDiving: A Fine-grained Dataset for Procedure-aware Action Quality Assessment
MIT License
113 stars 11 forks source link

On the setting of batch size in test_loader #11

Open Lyman-Smoker opened 1 year ago

Lyman-Smoker commented 1 year ago

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:

image

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!