Out of memory while testing

[Bouncer51]

hey,
while i was running test_pgn.py on the sample images provided. i sometimes get warnings like cannot allocate memory but it runs slowly.
can I change anything like batch size while running test_pgn.py

I use gtx 1050 gpu.