Can you please post your inference code or snippet? It shouldn’t give CUDA OOM error for resize operation. Also, can you try with smaller image size e.g., 64 and check whether you are getting CUDA OOM error?