search

AI4Bharat / IndicTrans2

Translation models for 22 scheduled languages of India
https://ai4bharat.iitm.ac.in/indic-trans2
MIT License
214 stars 59 forks source link

Hardware Requirement #66

Closed datha29 closed 5 months ago

datha29 commented 5 months ago

Hi I am trying to fine tune the model.Despite using 8 GPU there is an out of memory issue shown.Would like to know resolution for the same

PranjalChitale commented 5 months ago

Please adjust your batch size according to your configuration to mitigate the OOM issue.