Lesson 1 - Unable to train IMDB sentiment on Google Colab

I am trying to train the IMDB model from Lesson 1, and so far I have tried:

  • GPU - default batch size, bs=32, bs=16
  • TPU - default batch size, bs=32
    and it keeps either running out of memory or timing out. Has anyone else experienced this?
    Specifically, the code I am running is:
!pip install -Uqq fastbook
import fastbook
from fastai.text.all import *
dls = TextDataLoaders.from_folder(untar_data(URLs.IMDB), valid='test')
learn = text_classifier_learner(dls, AWD_LSTM, drop_mult=0.5, metrics=accuracy)
learn.fine_tune(4, 1e-2)

This worked on a Colab GPU with bs=8. It took 1.5 hours to run. Is that to be expected?

I actually took out the paid Google Colab and that worked perfectly. I was joining IMDB with movielens and mine took six hours.
Regards Conwyn