I am following F.Chollet book "Deep learning with python" and can\'t get one example working. In particular, I am running an example from chapter "Training a conv