- 博客(1)
- 资源 (1)
- 收藏
- 关注
原创 tflite模型调试 - 中间层output输出
背景由于tflite模型设计时考虑到了内存占用等问题,其forward的过程会反复利用中间内存空间,而导致中间层的output数据没有保存下来,模型debug难点较大。经过查阅相关材料,发现一个最简洁的方式,记录于此。思路:解析flatbuffer格式的tflite模型, 直接修改tflite模型内容,调整模型output到指定层。实现1. 修改tflite模型output的函数。from tensorflow.lite.python import schema_py_genera
2021-01-12 15:38:22 3334 27
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人