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.

Sign up for GitHub

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

Jump to bottom

Simple ViT error #7

Open

matrix0117 opened this issue Oct 24, 2023 · 1 comment

matrix0117 commented Oct 24, 2023 •

edited

Loading

Hello, I'm trying to apply this to cv, using the code from Simple Vit give me this error.

There seems to be an error in this code

NishchalPrasad commented Mar 19, 2024

You need to squeeze the tensor in dimensions 1 and 2. It should work then.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment