【问题标题】:Can you use the training parameter with tf.keras.Model() constructor?您可以将训练参数与 tf.keras.Model() 构造函数一起使用吗?
【发布时间】:2020-01-08 03:59:01
【问题描述】:

是否有可能将training 参数的正确值传递给使用tf.keras.Model() 构造函数(“功能”API)制作的模型中的层,或者您是否必须继承tf.keras.Model() 才能做到这一点?

对于tf.keras.layers.BatchNormalization,我需要training

【问题讨论】:

    标签: python tensorflow deep-learning tf.keras


    【解决方案1】:

    简短回答:并非如此。
    Keras Layer 具有您可以访问和修改的 trainable 参数。但这更多地用于冷冻重量。

    至于Sequential API,就目前而言,关于 BatchNorm 在推理模式下的行为方式的文档很少。甚至还有一些已打开的 GitHub 问题 herehere

    tf.keras.Model() API 让您可以更好地控制流程以及层的行为方式,所以如果您不关心Saved Model 格式,我会坚持使用。

    【讨论】:

      猜你喜欢
      • 1970-01-01
      • 2014-03-10
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 2023-03-20
      • 1970-01-01
      • 1970-01-01
      • 2018-11-09
      相关资源
      最近更新 更多