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

bug fix in the function update_probs and cuda device issue in sparseconvunet and bug fix in cuda device issue in pointtransformer #644

Open
wants to merge 3 commits into
base: main
Choose a base branch
from

Conversation

sealion789
Copy link

@sealion789 sealion789 commented Mar 20, 2024

The two models used to error out when run in a docker and device is set to cuda with the following error. Now with these fixes the error no longer exist when device is set to cuda. The sparseconvu net model also had an extra position argument in update_probs() function that is removed now.
device_issue2
device_issue1

@sealion789 sealion789 changed the title bug fix in the function update_probs, to remove the positional argument test_labels bug fix in the function update_probs and cuda device issue in sparseconvunet and bug fix in cuda device issue in pointtransformer Jun 2, 2024
@ssheorey ssheorey requested review from ssheorey and benjaminum and removed request for ssheorey July 17, 2024 05:54
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
Status: No status
Development

Successfully merging this pull request may close these issues.

1 participant