zoukankan      html  css  js  c++  java
  • tensorrt中builder.max_workspace_size的作用

    首先单位是字节,比如 builder.max_workspace_size = 1<< 30 就是 2^30 bytes 即 1 GB。

    它的作用是给出模型中任一层能使用的内存上限。运行时,每一层需要多少内存系统分配多少,并不是每次都分 1 GB,但不会超过 1 GB。

    One particularly important property is the maximum workspace size.

    • Layer algorithms often require temporary workspace. This parameter limits the maximum size that any layer in the network can use. If an insufficient scratch is provided, it is possible that TensorRT may not be able to find an implementation for a given layer.

    一个特别重要的特性是最大工作空间大小。
    层算法通常需要临时工作空间。 此参数限制网络中任何层可以使用的最大大小。 如果提供的暂存不足,TensorRT 可能无法找到给定层的实现。

    参考:

    https://forums.developer.nvidia.com/t/the-significance-of-builder-max-workspace-size/72364/2

    https://docs.nvidia.com/deeplearning/tensorrt/developer-guide/index.html#build_engine_c

  • 相关阅读:
    对拍程序的写法
    单调队列模板
    [bzoj1455]罗马游戏
    KMP模板
    [bzoj3071]N皇后
    [bzoj1854][SCOI2010]游戏
    Manacher算法详解
    [bzoj2084][POI2010]Antisymmetry
    Python_sklearn机器学习库学习笔记(一)_一元回归
    C++STL学习笔记_(1)string知识
  • 原文地址:https://www.cnblogs.com/mrlonely2018/p/14841562.html
Copyright © 2011-2022 走看看