簡體   English   中英

您可以將訓練參數與 tf.keras.Model() 構造函數一起使用嗎?

[英]Can you use the training parameter with tf.keras.Model() constructor?

是否有可能獲得傳遞給使用tf.keras.Model()構造函數(“功能”API)制作的模型中的層的training參數的正確值,或者您是否必須tf.keras.Model()要做到這一點?

我需要training tf.keras.layers.BatchNormalization

簡短的回答:不是。
Keras 層具有trainable參數,您可以訪問和修改這些參數。 但這更多地用於凍結重量。

至於Sequential API,就今天而言,關於 BatchNorm 在推理模式下的行為的文檔很少。 這里這里甚至還有一些打開的 GitHub 問題。

tf.keras.Model() API 使您可以更好地控制流程和層的行為方式,因此如果您不關心Saved Model格式,我會堅持下去。

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM