gmm-torch

gmm-torch copied to clipboard

it will Raise RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu! if gmm.fix(x) be used in the second time. I dont konw hao to fix it. The code as gmm = GaussianMixture(n_components, x.size(1), covariance_type="diag").cuda() gmm.fit(x) gmm.fit(x) #this will raise error