Terapkan model di ml-engine, ekspor dengan tf.train.Saver()

Saya ingin menerapkan model pada versi baru Google ML Engine. Sebelumnya, dengan Google ML, saya dapat mengekspor model terlatih saya dengan membuat tf.train.Saver(), menyimpan model dengan saver.save(session, output).

Sejauh ini saya belum dapat mengetahui apakah model yang diekspor yang diperoleh dengan cara ini masih dapat diterapkan di mesin ml, atau saya harus mengikuti prosedur pelatihan yang dijelaskan di sini dan buat paket pelatih baru dan latih model saya dengan ml-engine.

Apakah saya masih dapat menggunakan tf.train.Saver() untuk mendapatkan model yang akan saya terapkan di mesin ml?


google-cloud-platform google-cloud-ml-engine
person EffePi    schedule 21.03.2017    source sumber

Jawaban (1)


arrow_upward
4
arrow_downward

tf.train.Saver() hanya menghasilkan pos pemeriksaan.

Cloud ML Engine menggunakan SavedModel, yang dihasilkan dari API berikut: https://www.tensorflow.org/versions/master/api_docs/python/tf/saved_model?hl=bn

Model yang disimpan adalah pos pemeriksaan + protobuf berseri yang berisi satu atau lebih definisi grafik + sekumpulan tanda tangan yang menyatakan masukan dan keluaran grafik/model + file aset tambahan jika berlaku, sehingga semua ini dapat digunakan pada waktu penayangan.

Saya sarankan melihat beberapa contoh:

  1. Contoh sensus - https://github.com/GoogleCloudPlatform/cloudml-samples/blob/master/census/tensorflowcore/trainer/task.py#L334

  2. Dan contoh/kode perpustakaan saya sendiri - https://github.com/TensorLab/tensorfx/blob/master/src/training/_hooks.py#L208 yang memanggil ke https://github.com/TensorLab/tensorfx/blob/master/src/prediction/_model.py#L66 untuk mendemonstrasikan cara menggunakan sebuah pos pemeriksaan, memuatnya ke dalam sesi dan kemudian menghasilkan model yang disimpan.

Semoga petunjuk ini membantu mengadaptasi kode Anda yang ada untuk menghasilkan model yang sekarang menghasilkan SavedModel.

Saya rasa Anda juga menanyakan pertanyaan serupa lainnya untuk mengonversi model yang diekspor sebelumnya, dan saya akan menautkannya di sini untuk kelengkapan bagi orang lain: Menerapkan inception SavedModel yang telah dilatih ulang ke mesin google cloud ml

person Nikhil Kothari    schedule 27.03.2017