search

sanchit-gandhi / seq2seq-speech

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

Convert Flax training script to a standalone training script #5

Closed sanchit-gandhi closed 2 years ago

sanchit-gandhi commented 2 years ago

Import the FlaxSpeechEncoderDecoderModel from the seq2seq repository instead of HF Transformers.