mozilla / firefox-translations-training

Training pipelines for Firefox Translations neural machine translation models
https://mozilla.github.io/firefox-translations-training/
Mozilla Public License 2.0
142 stars 31 forks source link

Improve GPU utilization for "translate" tasks #785

Open eu9ene opened 1 month ago

eu9ene commented 1 month ago

Currently, it's ~70%. We could try using a bigger batch but it also depends on language.

GCP console for translate-mono task)

Screenshot 2024-07-31 at 2 30 12 PM
eu9ene commented 1 month ago

It appears to be even lower for translate-corpus: GCP console

Screenshot 2024-07-31 at 2 36 06 PM

@gregtatum FYI

gregtatum commented 1 month ago

Is it possible to dynamically determine this value? Like run N translations, measure and adjust?