torch.cuda.OutOfMemoryError =========================== .. currentmodule:: torch.cuda .. autoexception:: OutOfMemoryError