search

Shivanandroy / simpleT5

simpleT5 is built on top of PyTorch-lightning⚡️ and Transformers🤗 that lets you quickly train your T5 models.
MIT License
386 stars 62 forks source link

Model.predict on vector of strings #29

Open Nagakiran1 opened 2 years ago

Nagakiran1 commented 2 years ago

Thanks for outsourcing nice repo @Shivanandroy did you also develop functionality to predict multiple strings as a single vector like batch execution. Its computationally expensive to predict one by one

entropy2333 commented 2 years ago

Thanks for outsourcing nice repo @Shivanandroy did you also develop functionality to predict multiple strings as a single vector like batch execution. Its computationally expensive to predict one by one

You can refer to #42