Getting CUDA OOM on lesson3-imdb notebook with a bs=8

If you reload the data_lm (ie: after OOM error), you should specify new “bs” even in TextLMDataBunch.load function. Probably load function uses default bs (=64)…
To make it works in 8GB board I should add “bs=32”:

data_lm = TextLMDataBunch.load(path, 'tmp_lm', bs=32)

4 Likes