SparseTT issues

RuntimeError: CUDA error: CUBLAS_STATUS_NOT_SUPPORTED when calling `cublasSgemm( handle, opa, opb, m, n, k, &alpha, a, lda, b, ldb, &beta, c, ldc)`

Test error: File "/gpfs/work/mec/junweiwu20/SparseTT-main/videoanalyst/model/backbone/backbone_impl/swin_transformer.py", line 94, in forward qkv = self.qkv(x).reshape(B, N, 3, self.num_heads, File "/gpfs/work/mec/junweiwu20/anaconda3/envs/SparseTT/lib/python3.7/site-packages/torch/nn/modules/module.py", line 1102, in _call_impl return forward_call(*input, **kwargs) File "/gpfs/work/mec/junweiwu20/anaconda3/envs/SparseTT/lib/python3.7/site-packages/torch/nn/modules/linear.py", line 103, in forward return...

jasonwjw

mmcv

1

请问mmcv这个包需要什么版本的合适，如何安装？requirements里给的是1.5.0版本，但是显示不兼容

wsasdsda

精度差别太大

4

复现SparseTT，只训练got10k，测试的结果只有0.680左右，用了不同的电脑都是这样，差别很大。

ElliottZhen

请问推理部分是这么聚合两个头的结果的？

1

是要用到其他结构吗？

llz-lian

Welcome update to OpenMMLab 2.0

# Welcome update to OpenMMLab 2.0 I am Vansin, the technical operator of OpenMMLab. In September of last year, we announced the release of OpenMMLab 2.0 at the World Artificial...

vansin