如何在keras子类化模型中使用回调？

Question

我将keras子类模型实例化为tensorflow guide，

要能够将.fit用于我的模型类，我需要创建compute_output_shape。不过，对拟合模型使用回调会抛出NotImplementedError。

那么，如何在keras子类化模型中使用回调，例如张量板，检查点等？

Answer 1

您可以尝试以下方法：

model = SubclassModel()
# Callbacks you can define
tensorboard_callback = tf.keras.callbacks.TensorBoard(log_dir='model')
# Add callbacks to fit parameter
model.fit(data, labels, batch_size=100, epochs=5, callbacks=[tensorboard_callback])