Tensorflow中在继承tf.keras.layer.Layer
进行自定义层,并且将该层添加到tf.keras.Model
之后,程序可以正常运行,但Tensorboard中GRAPHS
页面下Tag
中只有Default
图没有keras
图,是所有的包含自定义层的Model都不会显示Graph
图吗,否则该如何修改,有什么注意事项。

Tensorflow自定义层后Tensorboard中GRAPHS页面下Tag中只有Default图没有keras图
- 写回答
- 好问题 0 提建议
- 关注问题
- 邀请回答
-
3条回答 默认 最新
- Ly.Leo 2023-03-18 17:41关注
其实原因很简单,如果模型中包含自定义层(或者其他子类化类可能也适用),需要实现
get_config
方法,之后一切都正常了本回答被题主选为最佳回答 , 对您是否有帮助呢?解决 无用评论 打赏 举报