• Eran Hirsch's avatar
    Add logits_processor parameter, used by `generate`, to `Seq2SeqTrainer`... · 13570381
    Eran Hirsch authored
    Add logits_processor parameter, used by `generate`, to `Seq2SeqTrainer` methods `evaluate` and `predict` (#17805)
    
    * Add logits_processor parameter, used by `generate`, to `Seq2SeqTrainer` methods `evaluate` and `predict`
    
    * Add all generate parameters to `Seq2SeqTrainer`, and also to `QuestionAnsweringSeq2SeqTrainer` which overrides it
    
    * Remove `self._num_beams` from trainer classes
    
    * - Run fixup
    - Fix "Constraint" not exposed
    - Fix synced_gpus to actually read from param
    
    * Use kwargs
    
    * Copy kwargs before making changes to it
    
    * Fix style issues unused imports
    13570381
trainer_seq2seq_qa.py 5.42 KB