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
135 stars 28 forks source link

Bump memory for shortlist #685

Closed eu9ene closed 1 week ago

eu9ene commented 1 week ago

It seems there are some OOMs for shortlist and unlike alignments, it's based on a proper sentence piece tokenization. There might be something else in the Python code and I'm investigating it now but in the meantime let's bump memory for it too.