sanchit-gandhi / seq2seq-speech

Repository for fine-tuning Transformers 🤗 based seq2seq speech models in JAX/Flax.
34 stars 6 forks source link

[Seq2Seq Train] Beam at each eval steps, greedy at end of epoch #51

Closed sanchit-gandhi closed 2 years ago

sanchit-gandhi commented 2 years ago

Perform greedy at end of epoch, beam at each eval steps and final eval (whether evaluating at every epoch or after eval steps).