SVD failed to converge on some samples

[guizilaile23]

Hi, Thanks for the the implementation, it's make me understand the paper more easily.
but during the transfer, I found some samples will make SVD failed.
" RuntimeError: Lapack Error gessd : 1 superdiagonals failed to converge. at /opt/conda/....."
and I already change " torch.svd(c_covm, some= True)", but still got this error, you have any idea to deal with that?
Thanks