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

Fix use_cuda of swin transformer example code #505

Open
wants to merge 1 commit into
base: master
Choose a base branch
from

Conversation

chilin0525
Copy link

Based on #504, use_cuda already removed in 00711a2, so user will get error when executing swin transformer example code currently.

After fixing, the code will run successfully for CPU and GPU scenario, following is my test result:

$ python3 usage_examples/swinT_example.py --use-cuda
Using GPU for acceleration
100%|████████████████████████████████████████████████████████████████████████████████████████████████████████| 32/32 [00:07<00:00,  4.43it/s]

$ python3 usage_examples/swinT_example.py
Using CPU for computation
100%|████████████████████████████████████████████████████████████████████████████████████████████████████████| 32/32 [02:01<00:00,  3.78s/it]

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

Successfully merging this pull request may close these issues.

None yet

1 participant