前言 – 阿嬤碎碎念
這篇文章將教大家如何使用 tf.GradientTape 和 tf.function Decorator 裝飾器,同樣是上次的 MNIST ,這次我們使用進階的寫法。請跟隨程式碼上的註解閱讀理解,並可以至 Github 上將Jupyter Notebook 程式碼下載下來,實際跑跑看。
Github 程式碼
重要觀念
- GradientTape
- training 基本流程
- tf.function (decorator)
- training data shape
這篇文章將教大家如何使用 tf.GradientTape 和 tf.function Decorator 裝飾器,同樣是上次的 MNIST ,這次我們使用進階的寫法。請跟隨程式碼上的註解閱讀理解,並可以至 Github 上將Jupyter Notebook 程式碼下載下來,實際跑跑看。