search

Shivanandroy / simpleT5

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

Train on specific cuda device #32

Closed iknoorjobs closed 2 years ago

iknoorjobs commented 2 years ago

How to train on specific Cuda device?