Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Mamba.__init__() got an unexpected keyword argument layer_idx #3

Open
Shadow581 opened this issue Aug 25, 2024 · 8 comments
Open

Mamba.__init__() got an unexpected keyword argument layer_idx #3

Shadow581 opened this issue Aug 25, 2024 · 8 comments

Comments

@Shadow581
Copy link

Hello, I has followed your instruction to install env, but it shows errors.

link: SkyworkAI/PointCloudMamba#9

This problem bothered me for a long time.

@xiaoyao3302
Copy link
Owner

Hi, thanks for your interest in our work. Did you come across this problem when running our code?

@Shadow581
Copy link
Author

Thanks your reply.

I followed your instruction to install. (except python, torch and cuda version)

image

Execute your code, but it shows errors about knn_cuda Lib.
image

My cuda is available:
image

@xiaoyao3302
Copy link
Owner

Is CUDA version right? Does it match the knn dependencies? I remembered that installing knn and FPS requires a determined CUDA version. BTW what does GPT say?

@Shadow581
Copy link
Author

Shadow581 commented Aug 25, 2024

Is CUDA version right? Does it match the knn dependencies? I remembered that installing knn and FPS requires a determined CUDA version. BTW what does GPT say?

image
Is this the wrong match? I am using a cloud platform and I didn't configure the torch and cuda versions

I only have one cuda version, cuda-11.8.

image

I following advices of GPT, but it still shows errors.

@Shadow581
Copy link
Author

Additionally, I run my own code on the same env, it doesn't show errors. It can get knn data.
image
image
image

@Shadow581
Copy link
Author

Is CUDA version right? Does it match the knn dependencies? I remembered that installing knn and FPS requires a determined CUDA version. BTW what does GPT say?

Hello, I sovle the problem. Set CUDA_VISIBLE_DEVICES=0 in run.sh file.

@Shadow581
Copy link
Author

Hello, everyone, I have sovled the problem stated in the title.
I configured Mamba to just copy the models/block.py/Block code into my code, but this gave me an error(such as title). You need to copy all the functions about wrapping Mamba code into your code (up to the MixerModel class).

@xiaoyao3302
Copy link
Owner

Cool

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants