Very similar to this file with **Keras**: https://github.com/tensorlayer/tensorlayer/blob/master/example/tutorial_keras.py We should create the same thing with TF-Slim