运行 热切地在 Keras 中建模
Running models eagerly in Keras
我注意到即使在使用 .compile(..., run_eagerly=False)
编译模型后,.call()
中的打印语句仍然有效。这是否意味着 .call()
将手动包装在 tf.function
中?
run_eagerly
在 compile()
中影响属性 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()
.
更高级的逻辑
我注意到即使在使用 .compile(..., run_eagerly=False)
编译模型后,.call()
中的打印语句仍然有效。这是否意味着 .call()
将手动包装在 tf.function
中?
run_eagerly
在 compile()
中影响属性 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()
.