google-research / google-research/language

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

Open
#179 0 comments 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
1.8k
Forks
362
PR merge metrics
No merged PRs in 30d

Description

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?

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.