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

The question about imagenet feature #15

Open
stein-666 opened this issue Jan 13, 2025 · 1 comment
Open

The question about imagenet feature #15

stein-666 opened this issue Jan 13, 2025 · 1 comment

Comments

@stein-666
Copy link

Hi, thanks for your awesome work.
When I load the ImageNet feature downloaded from Hugging Face, an error is raised stating that the number of features does not equal 1,281,167. I wonder if I made a mistake when processing the data.

@YuchuanTian
Copy link
Owner

Hi, thank you for your interest in our work!
How many cards (subprocesses) have you inited for training? If the card number is less than 8, the error might be raised. A solution is to reallocate the batchsize dimension, e.g. loading several numpy in one process and concat them on the 0th dimension.

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