gonbaez

@piyushongithub I had the same issue. Apparently is because the order of the weights changes when saving the model with modified trainable attributes [link](https://datascience.stackexchange.com/questions/49546/keras-load-pre-trained-weights-shape-mismatch/). I fixed it by saving the...