Saving and Loading saved models gives different results #24

Hi,
There is a serious problem in the current codebase. If you save a model then reload it in a DIFFERENT time (not the same execution of main.py) the accuracy is 50% on IMDB. As a sanity check if you save and reload in the execution of main.py then there is no problem. If I had to guess this is a dataloading problem where there is a mistmatch between the saved model and the newly loaded model in a second execution.