Ly.Leo 2023-02-22 15:04 采纳率: 80%
浏览 30
已结题

Tensorflow自定义层后Tensorboard中GRAPHS页面下Tag中只有Default图没有keras图

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

  • 写回答

3条回答 默认 最新

  • Ly.Leo 2023-03-18 17:41
    关注

    其实原因很简单,如果模型中包含自定义层(或者其他子类化类可能也适用),需要实现get_config方法,之后一切都正常了

    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论
查看更多回答(2条)

报告相同问题?

问题事件

  • 系统已结题 3月26日
  • 已采纳回答 3月18日
  • 修改了问题 2月22日
  • 创建了问题 2月22日