让 TensorBoard 支持具有自定义子类的 Keras 模型

Keras 中的TensorBoard回调可以方便的用来记录模型训练过程中的各种参数变化，以在训练后对模型进行分析。在 Keras 的官方示例中，展示了如何将TensorBoard回调用在自定义的 Keras 模型中。但是在实际使用中，如果你像我一样继承了写 PyTorch 的习惯，将整个 Keras 模型模块化成多个子模型的话，就会发现TensorBoard回调在这个时候失效了。这个时候就需要对TensorBoard回调的功能做增强，让它支持 Keras 模型中的子模型。