kennymckormick / pyskl

A toolbox for skeleton-based action recognition.
Apache License 2.0
911 stars 175 forks source link

Adjusting batch size for demo_skeleton.py #199

Open sjwang222 opened 10 months ago

sjwang222 commented 10 months ago

Hi Kenny,

Thank you for your wonderful work. I believe I have successfully installed this package on my computer following your instruction. While I was playing with the demo, the terminal reported 'RuntimeError: CUDA out of memory. Tried to allocate xxx...'. I think my GPU is a little bit out of date and I am wondering whether I can reduce the batch size in 'demo_skeleton.py' to get it work (but I cannot find any keyword like 'batch' or 'size' therein). Any hints will be appreciated!

Thanks! Ryan