張量流會議

Question

我是張量流初學者。所以，請原諒我的問題，如果它是愚蠢的

我檢查了使用MNIST數據和tensorflow實現CNN的github代碼。 下面的鏈接：

https://github.com/aymericdamien/TensorFlow-Examples/blob/master/examples/3_NeuralNetworks/convolutional_network.py

不過，我需要保存這段代碼生成的模型，但不知道如何做到這一點，因爲這代碼不涉及使用會話，如何在它上面運行會話？

希望你的迴應。

Answer 1

鏈接代碼使用tf.estimator.Estimator來訓練模型。其documentation包括如何使用export_savedmodel保存模型。保存的模型可以通過tf.estimator.Estimator初始值的model_dir參數指定其位置來導入。