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)