CosmoStat / wf-psf

Data-driven wavefront-based PSF modelling framework.
MIT License
19 stars 9 forks source link

GPU usage below 50% when training the model #69

Open tobias-liaudat opened 1 year ago

tobias-liaudat commented 1 year ago

This should be investigated. I suspect that some operations are being done in CPU which slows the training and underutilises the GPU.

If my suspicion is correct, identifying the operations that slow down the training on the model and making sure they're run on GPU will speed up the training.