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)