Should I change other parameters when I change batch_size?

ttengwang / PDVC

End-to-End Dense Video Captioning with Parallel Decoding (ICCV 2021)

MIT License

200 stars 23 forks source link

Should I change other parameters when I change batch_size? #23

Closed RDCJ closed 2 years ago

RDCJ commented 2 years ago

I cloned your repository and trained the model on ActivityNet. When I changed batch_size to a higher number, there was an error in the program. It happened at line 387 of pdvc.py, and after debugging, I found that the program ran there with an out-of-bounds in tensor. Should I change other parameters when I change batch_size?

ttengwang commented 2 years ago

See https://github.com/ttengwang/PDVC/issues/4

ttengwang commented 2 years ago

@RDCJ But if you really need a version with batch size >1, I will update the code in my spare time asap.