Skip to content

Commit

Permalink
Merge pull request #263 from Redmept1on/master
Browse files Browse the repository at this point in the history
Update 13-tensorflow.md
  • Loading branch information
sparanoid authored Jul 17, 2024
2 parents 4f82a5d + 13e1a92 commit 47bc273
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion docs/arch/arch/13-tensorflow.md
Original file line number Diff line number Diff line change
Expand Up @@ -34,7 +34,7 @@ TVM 对动态张量 shape 的支持最少。为 `None` 的维度应替换为常

### 导出

TensorFlow 前端需要一个冻结的 protobuf(.pb)或保存的模型作为输入。它目前不支持检查点(.ckpt)。TensorFlow 前端所需的 graphdef 可以从活动 session 中提取,或者使用 [TFParser](https://github.com/apache/tvm/blob/main/python/tvm/relay/frontend/tensorflow_parser.py) 辅助类。
TensorFlow 前端需要一个冻结的 protobuf(.pb)或保存的模型作为输入。它目前不支持检查点(.ckpt)。TensorFlow 前端所需的 graphdef 可以从会话 session 中提取,或者使用 [TFParser](https://github.com/apache/tvm/blob/main/python/tvm/relay/frontend/tensorflow_parser.py) 辅助类。

导出模型时应进行一些转换,以准备模型进行推理。设置 `add_shapes=True` 也很重要,因为这会将每个节点的输出 shape 嵌入到图中。这是一个将模型导出为给定会话的 protobuf 的函数:

Expand Down

0 comments on commit 47bc273

Please sign in to comment.