• Nayan Singhal's avatar
    Alignment Training task using minibatch · 2c278ff0
    Nayan Singhal authored
    Summary:
    1. Define a EpochMinibatchIterator which extends the EpochBatchIterator. It has same functionality as EpochBatchIterator except two major changes: use static batching and use MiniBatchIterator for getting the indices.
    
    2. SplitSeqCollater is used instead of Seq2SeqCollater.
    3. LSTM_subsample started storing the previous states and reset it once the sample is over.
    
    Reviewed By: jay-mahadeokar
    
    Differential Revision: D15209023
    
    fbshipit-source-id: 900b8bd1f25159ffc77f8106e26729a3e7422a1f
    2c278ff0
fairseq_model.py 12 KB