Adjusting batch size for demo_skeleton.py

[sjwang222]

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