def METHOD_NAME(self, preds: Tensor, target: Tensor) -> None:
    """Update state with predictions and targets."""
    if self.rmse_map is None:
        _img_shape = target.shape[1:]  # channels, width, height
        self.rmse_map = torch.zeros(_img_shape, dtype=target.dtype, device=target.device)
    self.rmse_val_sum, self.rmse_map, self.total_images = _rmse_sw_update(
        preds, target, self.window_size, self.rmse_val_sum, self.rmse_map, self.total_images
    )