gpt4 book ai didi

tensorflow - TF-TRT 与 UFF-TensorRT

转载 作者:行者123 更新时间:2023-12-03 14:48:04 26 4
gpt4 key购买 nike

我发现我们可以通过多种方式优化 Tensorflow 模型。如果我弄错了,请告诉我。

1- 使用 TF-TRT , 这个 API 开发者由 tensorflow 和集成 TensoRT 到 Tensorflow 和这个 API 称为:

from tensorflow.python.compiler.tensorrt import trt_convert as trt



这个 API 可以应用于任何 tensorflow 模型(新旧版本模型),没有任何转换错误,因为如果这个 API 不支持任何新层,那么 TensorRT 引擎不要考虑这些层,而这些层保留给 Tensorflow 引擎和在 Tensorflow 上运行。对?

2- 使用 TensorRT , 这个 API 由 NVIDA 开发,独立于 Tenorflow 库(未集成到 Tensorflow),这个 API 称为:

import tensorrt as trt



如果我们要使用这个api,首先我们必须使用uff-convertor将tensorflow graph转换为UFF,然后将UFF graph解析为这个API。
在这种情况下,如果 Tensorflow 图有不受支持的层,我们必须为这些层使用插件或自定义代码,对吗?

3- 我不知道,当我们使用 Tensorflow 模型时,为什么我们使用 UFF 转换器然后使用 TensorRT,我们可以直接使用 TF-TRT API,对吗?如果是这样,您是否通过这两种方法测试了 Tensorflow 优化模型以获得相同的性能?这种 UFF 转换器方法的优点是什么?

我对上述两种情况有一些疑问:

4- 我使用两种情况转换 ssd_mobilenet_v2,在情况 1 中,我的速度略有提高,但在情况 2 中,我取得了更大的进步,为什么?
我的观点是,在情况 1 中,API 只考虑转换精度(FP32 到 FP16)并将可能的层合并在一起,但在情况 2 中,图形通过 UFF 是干净的,例如删除任何冗余节点,如断言和身份然后转换为张量图,对吗?

5- 当我们转换训练好的模型文件时,如 .ckpt.meta , ... 到卡住推理图( .pb file ),这些层不会从图中删除?只有损失状态和优化器状态,​​...被删除?

最佳答案

此处有答案的重复帖子:https://github.com/NVIDIA/TensorRT/issues/341

关于tensorflow - TF-TRT 与 UFF-TensorRT,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/59785769/

26 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com