TensorFlow 2 教學:Keras–MNIST–自訂模型

tensorflow

前言阿嬤碎碎念

這篇文章將教大家如何使用 tf.GradientTape 和 tf.function Decorator 裝飾器,同樣是上次的 MNIST ,這次我們使用進階的寫法。請跟隨程式碼上的註解閱讀理解,並可以至 Github 上將Jupyter Notebook 程式碼下載下來,實際跑跑看。

Github 程式碼

重要觀念

  1. GradientTape
  2. training 基本流程
  3. tf.function (decorator)
  4. training data shape

Jupyter Notebook 程式碼

留言討論區