ModuleNotFoundError: No module named 'Swin3D.sparse_dl.attn_cuda'
Nobody has claimed this yet.
- Dominant language
- Cuda
- Stars
- 298
- Forks
- 26
- Avg merge
- 22h 44m
- Merged PRs (30d)
- 1
Description
thanks for your work~!
but when i run:
import torch
from Swin3D.models import Swin3DUNet
model = Swin3DUNet(depths, channels, num_heads,
window_sizes, quant_size, up_k=up_k,
drop_path_rate=drop_path_rate, num_classes=num_classes,
num_layers=num_layers, stem_transformer=stem_transformer,
upsample=upsample, first_down_stride=down_stride,
knn_down=knn_down, in_channels=in_channels,
cRSE='XYZ_RGB_NORM', fp16_mode=1)
model.load_pretrained_model(ckpt_path)
i got this bug: ModuleNotFoundError: No module named 'Swin3D.sparse_dl.attn_cuda'
emmm, what can i do???
Contributor guide
No contributing guide indexed for this repository
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Research direction
Start by reproducing the import failure from the shown Swin3DUNet example and inspect how Swin3D.sparse_dl.attn_cuda is expected to be installed or built. Done means the module imports successfully and the displayed model initialization can proceed without ModuleNotFoundError.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- python, pytorch
- Domain
- machine-learning
- Issue type
- Bug
- Difficulty
- 4/5
- Estimated time
- 3-5 days
- Activity status
- Stale
- Clarity
- Needs clarification
- Newbie friendliness
- 25/100