运行 热切地在 Keras 中建模

Running models eagerly in Keras

我注意到即使在使用 .compile(..., run_eagerly=False) 编译模型后,.call() 中的打印语句仍然有效。这是否意味着 .call() 将手动包装在 tf.function 中?

run_eagerlycompile() 中影响属性 tf.keras.Model.run_eagerly,但不在 call() 中使用。因为,call() 是 tf.keras.Model 中的 NotImplemented,它用于自定义训练模式,而不是“comple() fit()”编码模式。您可以在中找到 tf.function 包装逻辑 tf.keras.Model.make_train_function,由fit()自动包装train_step()等,这是比call().

更高级的逻辑