roerich icon indicating copy to clipboard operation
roerich copied to clipboard

ChangePointDetectionRuLSIF does not work on GPU

Open avnikitin opened this issue 11 months ago • 0 comments

The following code

from roerich.change_point import ChangePointDetectionRuLSIF

alg = ChangePointDetectionRuLSIF()
alg.predict(np.zeros((500, 3))) # just for example

throws an error: RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cpu and cuda:0! (when checking argument for argument mat1 in method wrapper_CUDA_addmm)

It works well on CPU, but when CUDA is available, the error appears.

avnikitin avatar Jul 31 '23 22:07 avnikitin