google-research / language

Shared repository for open-sourced projects from the Google AI Language team.
https://ai.google/research/teams/language/
Apache License 2.0
1.58k stars 341 forks source link

ORQA fine tuning with REALM pre-training: is it possible to run the fine tuning on Google Colab GPU? #179

Open turjo-001 opened 8 months ago

turjo-001 commented 8 months ago

Since I lack the hardware to run the ORQA fine tuning code on local machine, I was trying to run the fine tuning code on Colab using its GPU runtime. The thing I noticed is that Conda workaround I used for colab does not use colab's GPU resource at all. Even then, I tried to run the the code for fine tuning and got error after some time.

I managed to run the REALM pre training with my own dataset and wanted to do fine tune ORQA to see its performance with my REALM pretraining. My question now is, is it even possible to run ORQA fine tuning on Colab? Are there any workarounds or things I'm missing?