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.

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

RuntimeError: shape mismatch: value tensor of shape [1152, 896] cannot be broadcast to indexing result of shape [392, 896] #3

Open
sugarandgugu opened this issue Dec 2, 2024 · 1 comment
Open

RuntimeError: shape mismatch: value tensor of shape [1152, 896] cannot be broadcast to indexing result of shape [392, 896] #3

sugarandgugu opened this issue Dec 2, 2024 · 1 comment

Comments

@sugarandgugu
Copy link

SigLIP下载应该是这个模型google/siglip-base-patch16-224，如果是384利用github的源码shape是对不上的，224能正常跑。
下载模型使用命令：
export HF_ENDPOINT=https://hf-mirror.com
huggingface-cli download --resume-download google/siglip-base-patch16-224 --local-dir ./siglip-patch16-224
@wyf3
Copy link
Owner

wyf3 commented Dec 5, 2024

SigLIP下载应该是这个模型google/siglip-base-patch16-224，如果是384利用github的源码shape是对不上的，224能正常跑。 下载模型使用命令： export HF_ENDPOINT=https://hf-mirror.com huggingface-cli download --resume-download google/siglip-base-patch16-224 --local-dir ./siglip-patch16-224

谢谢提醒，我训练的时候使用的是224的，在README中写错了，已经修改

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@sugarandgugu @wyf3