search

wentaoyuan / deepgmr

PyTorch implementation of DeepGMR: Learning Latent Gaussian Mixture Models for Registration (ECCV 2020 spotlight)
https://wentaoyuan.github.io/deepgmr
Other
145 stars 15 forks source link

svd did not converge #7

Open chalth opened 3 years ago

chalth commented 3 years ago

Hello, when i tried to train the model, i meet with this error. Do you konw how to deal with this?

Traceback (most recent call last): File "train.py", line 174, in train_one_epoch(epoch, model, train_loader, writer, args.log_freq, args.plot_freq) File "train.py", line 41, in train_one_epoch loss, r_err, t_err, rmse = model(pts1, pts2, T_gt) File "/home/lthach/anaconda3/envs/mink/lib/python3.7/site-packages/torch/nn/modules/module.py", line 727, in _call_impl result = self.forward(*input, **kwargs) File "/home/lthach/deepgmr-master/model.py", line 177, in forward self.T_12 = gmm_register(self.pi1, self.mu1, self.mu2, self.sigma2) File "/home/lthach/deepgmr-master/model.py", line 46, in gmmregister U, , V = torch.svd(Ms) RuntimeError: svd_cuda: the updating process of SBDSDC did not converge (error: 2)

Rancho86 commented 3 years ago

I meet the same error

liu-zikang commented 2 years ago

Have you already solved the question？I meet the same error too.