I’m looking at trying to improve the robustness of our trainer code.  If I select batch parameters that are not tight enough, I may run out of memory.  Is it safe to catch the OOM, reduce the batch size, and try again?
Thanks
Jerry