Get_data function running out of memory on training data in lesson 2

[mattobrien415]

@clu2033, you could try using fit_generator then