search

Gorilla-Lab-SCUT / MAST

[IJCAI 2023] Manifold-Aware Self-Training for Unsupervised Domain Adaptation on Regressing 6D Object Pose Installation
MIT License
4 stars 0 forks source link

question #1

Open AFESDTTM opened 5 months ago

AFESDTTM commented 5 months ago

How can I handle this essue.
File "/data/ss/MAST/MAST/models/loss_ops.py", line 270, in geodesic_dists internal = 0.5 * (torch.diagonal(torch.matmul(R_bin_ctrs, torch.transpose(R,-1, -2)), dim1=-1, dim2=-2).sum(-1) - 1.0) RuntimeError: CUDA error: CUBLAS_STATUS_EXECUTION_FAILED when calling cublasSgemm( handle, opa, opb, m, n, k, &alpha, a, lda, b, ldb, &beta, c, ldc)

aragakiyui611 commented 1 month ago

I didn't meet this problem before, try install pytorch==1.8.1