Forming averaged marian models?

32 views
Skip to first unread message

sjbr...@gmail.com

unread,
Oct 12, 2021, 3:50:09 PM10/12/21
to marian-nmt
Is there a tool already existing that will combine multiple, equivalently parameterized Marian models, to find the averaged model? 

Averaged models have proven very useful w/ other tools, Sockeye, for example.

Thanks,

Steve

Jindra Helcl

unread,
Oct 14, 2021, 12:01:03 PM10/14/21
to marian-nmt
Hello Steve,

there is a script in the Marian repository which should do exactly what you need: https://github.com/marian-nmt/marian/blob/master/scripts/checkpoints/average.py 

Best,
Jindra

Dne úterý 12. října 2021 v 20:50:09 UTC+1 uživatel sjbr...@gmail.com napsal:

sjbr...@gmail.com

unread,
Oct 15, 2021, 6:51:39 PM10/15/21
to marian-nmt
Thanks! perfect.
Reply all
Reply to author
Forward
Message has been deleted
0 new messages