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

Simple modification suggestion for the requirments.txt #470

Open
jonghyunlee1993 opened this issue Oct 4, 2024 · 0 comments
Open

Simple modification suggestion for the requirments.txt #470

jonghyunlee1993 opened this issue Oct 4, 2024 · 0 comments

Comments

@jonghyunlee1993
Copy link

Hi!

I'm really thanks to this amazing project and source code.

I encountered a strange error during the pretraining run, which appeared to occur when adjusting the hue value. I suspect the issue was caused by the augmentation process in torchvision.

Specifically, the error was an overflow occurring in a numpy array, which was triggered by using numpy > 2.0. One solution is to upgrade torchvision to version 0.19 or higher, but this isn't feasible because it conflicts with the requirements for torch==2.0.0 and xformer. As a workaround, I downgraded numpy to a version below 2.0, specifically 1.26.4.

It took quite a bit of trial and error to identify and fix the issue, so it would be helpful to add a numpy version restriction to the requirements.txt to prevent similar problems in the future.

I referenced the following article: pytorch/vision#8536

Thanks.

@jonghyunlee1993 jonghyunlee1993 changed the title Suggestion for requirments.txt Simple modification suggestion for the requirments.txt Oct 4, 2024
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

1 participant