-
Notifications
You must be signed in to change notification settings - Fork 44
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
Extractor feature OOM #9
Comments
Hi @StarsTesla, The current PCA computation code extracts features from each image using the GPU. AFAIK, There are two possible options to the OOM error you receive: (2) Your GPU manages to extract a single image in this manner and doesn't handle properly many extracted images (You receive the OOM error after extracting features for several images). In this case I would recommend: Hope this helps, and good luck! |
@ShirAmir |
Hi there, I tried the code, at the beginning , everything sames fine, but when I try to use the extractor on my own images, more specificly, extract feature of high resolution pciture and visualize the pca picture. I tried the code on 100 800 x 800 images, set the load_size=224 and stride=2, it seems fine, so the code maybe run the image separately?
How should I calculate the memory of GPU and Could the Extractor could modify to using on multiple GPUs?
The text was updated successfully, but these errors were encountered: