roerich
roerich copied to clipboard
ChangePointDetectionRuLSIF does not work on GPU
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.