Skip to content

Conversation

talonchandler
Copy link
Contributor

@talonchandler talonchandler commented Mar 18, 2025

Fixes #34.

I found that max_pool3d was the memory bottleneck, requesting more than 141 GB of our memory for daxi volumes (overfilling an H200).

Here I'm working around it by computing max_pool3d on the CPU. For smaller datasets, I found the CPU to be quite fast.

Related aside (should not block this merge): After removing this GPU memory bottleneck, I found that the next bottleneck is the fitting routine, which stalls on some beads. The tqdm progress bar shows steady progress on many beads, but very slow progress on some (likely not-very-gaussian) beads. @ieivanov I suspect that I'm not doing a good job filtering beads, and I'll chat with you about your routine for picking peak-fitting parameters.

@talonchandler talonchandler requested a review from ieivanov March 18, 2025 00:13
@mattersoflight mattersoflight added this to the Advanced Analysis milestone Aug 14, 2025
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.

characterize-psf fails on 37 GB daxi2 volume

2 participants