I am having some issues with GPU-RAM issues and hence needing to reduce batch size during testing. Is there a way I can do so? Thanks