https://raw.githubusercontent.com/tensorflow/hub/master/examples/image_retraining/retrain.py