search

smousavi05 / EQTransformer

EQTransformer, a python package for earthquake signal detection and phase picking using AI.
https://rebrand.ly/EQT-documentations
MIT License
301 stars 148 forks source link

issue about multiple GPUs #149

Closed Chouyuhin closed 1 year ago

Chouyuhin commented 1 year ago

hi,mousavi, first, thanks to EQTransformer! I got an issue : I use eqt on my own dataset without changing model structure or training code But when I run the code on multiple GPUs, the usage on every GPU is so low, and the time of one epoch is more than using one gpu,

image

can you give any suggestion please?

smousavi05 commented 1 year ago

As I mentioned before, this is due to the data reading and pre-processing that prevent taking full advantage of multiple GPU