First of all thanks for your great work!
The code you provided is very easy to read and use, that's awesome!
I've been encoutring a problem while trying to train the train_test.py for test or train. After all the VGG16 extraction when it reachs 99% it gives me this error " Allocation of 9707550720 exceeds 10% of free system memory. " while I am working on a Cloud Service with 36Gb RAM
