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

Yes that would be a good idea.