2

Google Cloud Datalabノートブックを使用してテンソルフローモデルをトレーニングしています。オフライン予測のために訓練されたモデルを保存したいと思います。しかし、私はモデルを保存する方法については無知です。テンソルフローモデルの保存方法を使用する必要がありますか、そうするためのdatalab/googleクラウドストレージ固有の方法はありますか?この点に関する助けがあれば、非常に感謝しています。Google datalabノートブックで訓練されたテンソルフローモデルをオフライン予測用に保存するにはどうすればよいですか?

答えて

2

テンソルフローモデルの保存方法は使用できますが、ローカルディスクではなくGoogle Cloud Storageバケットに保存することをおすすめします。ほとんどのテンソルフローメソッドは、ファイル名の代わりにgs://プレフィックスを使用してGoogle Cloud Storageパスを受け入れます。

私はSavedModelBuilderを使用することをお勧めします。現在、最もポータブルであるためです。ここに例があります:https://github.com/GoogleCloudPlatform/cloudml-samples/blob/master/flowers/trainer/model.py#L393

関連する問題