Gautam Malu

I am also facing the same issue. @1996scarlet any idea?

I will upload the data_prepration script by end of the day, you should be able to generate train.pkl and other files from that script.

I have updated the code, try now. Let me know if you face any problems. I am also updating the README.md, it will take some time probably by tomorrow.

Try with `fit` instead `fit_generator()` like: `model.fit(train_data[0], train_data[1], batch_size=batch_size, nb_epoch=nb_epoch, callbacks=[checkpoint, tb],#, lr_scheduler], validation_data=(val_data[0], val_data[1]), shuffle=True, initial_epoch=initial_epoch)` 8448 == number images to be cosindered in one epoch, actually with batch...