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

patchcore subsample_embedding is very slow #533

Closed
Vvdinosaur opened this issue Sep 1, 2022 · 1 comment
Closed

patchcore subsample_embedding is very slow #533

Vvdinosaur opened this issue Sep 1, 2022 · 1 comment

Comments

@Vvdinosaur
Copy link

Vvdinosaur commented Sep 1, 2022

Hi,
I have 2000+ images( img_size = 360x640) to train, it is very slow at subsample_embedding stage.
is there any method to accelerate this ?

@Vvdinosaur Vvdinosaur changed the title patchcore subsample_embedding is too slow patchcore subsample_embedding is very slow Sep 1, 2022
@samet-akcay
Copy link
Contributor

@Vvdinosaur, until we optimize the code, you could perhaps reduce the number of layers you are extracting. For example, you could only extract the features from 'layer3'. You could potentially play with coreset_sampling_ratio and num_neighbors parameters to improve the computation as well.

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

3 participants