Tiff writer doesn't place predictions on CPU during inference

[AjeyPaiK]

Here, the predictions are not placed on the CPU before being converted to numpy.