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

post-processing: removing bottleneck nanmean/max #208

Merged
merged 3 commits into from Jun 15, 2023

Conversation

miili
Copy link
Contributor

@miili miili commented Jun 14, 2023

A bottleneck is the usage of nanmean and nanmax.

This dirty quick fix is replacing those functions with fuctions from the library bottleneck (https://github.com/pydata/bottleneck).

For my usecase I sped up the inference by ~30%.

seisbench/models/base.py Outdated Show resolved Hide resolved
@yetinam
Copy link
Member

yetinam commented Jun 15, 2023

Hi @miili ,
thanks for the PR! Great to see this timing improvement. I left a note regarding the torch.cuda.synchronize call but once that's resolved I'm happy to merge.

@yetinam yetinam merged commit f875205 into seisbench:main Jun 15, 2023
13 checks passed
@yetinam
Copy link
Member

yetinam commented Jun 15, 2023

Thanks again for the contribution!

@yetinam yetinam added this to the v0.5 milestone Jul 6, 2023
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

Successfully merging this pull request may close these issues.

None yet

2 participants