Google mengatakan:masukkan deskripsi gambar di sini python tensorflow

0
user12585593 19 November 2020, 11:37

1 menjawab

Jawaban Terbaik

Sebagai peringatan, freeze_graph umumnya tidak digunakan lagi di TensorFlow 2.x. Anda harus menggunakan Model Tersimpan untuk fungsi yang sama di Tensorflow 2.x. Saya akan menjawab pertanyaan ini dari perspektif TensorFlow 1.x.

Sebelum memahami perbedaan keduanya, Anda perlu mengetahui bagaimana model TensorFlow dibentuk.

Setiap model TensorFlow terdiri dari struktur data grafik, yang berisi objek Operasi, yang merupakan unit komputasi, dan objek Tensor, yang merupakan unit data yang mengalir di antara keduanya.

Namun, grafik saja tidak cukup untuk melakukan sesuatu seperti inferensi. Saat Anda melatih model, model tersebut mempelajari dan mengoptimalkan serangkaian parameter unik untuk berbagai bagian grafik Anda.

File PB, kemudian, mencakup kedua bagian ini - grafik yang mewakili struktur model, dan parameter yang telah dipelajari model melalui pelatihan.

Jadi kembali ke pertanyaan awal - apa perbedaan antara write_graph dan freeze_graph?

  • write_graph menuliskan grafik model (struktur) ke dalam file PB.
    • Ini tidak memerlukan pelatihan apa pun, sehingga tidak menyertakan parameter apa pun yang mungkin telah dipelajari model.
  • freeze_graph mengambil parameter model terlatih dari pos pemeriksaan pelatihan dan menyimpannya juga ke file PB.
1
Clarence Leung 19 November 2020, 09:00