MemoryConfig构造函数
功能说明
MemoryConfig类的构造函数,用于配置系统内存使用方式。
函数原型
def __init__(self, atomic_clean_policy=0, static_memory_policy=0 )
参数说明
参数名 |
输入/输出 |
描述 |
---|---|---|
atomic_clean_policy |
输入 |
是否集中清理网络中所有atomic算子占用的内存,取值包括:
|
static_memory_policy |
输入 |
网络运行时使用的内存分配方式。
默认值是0。
说明:
|
返回值
返回MemoryConfig类对象,作为NPURunConfig的参数传入。
约束说明
无。
调用示例
1 2 3 4 5 |
from npu_bridge.npu_init import * ... mem_config = MemoryConfig(atomic_clean_policy=0, static_memory_policy=0) session_config=tf.ConfigProto(allow_soft_placement=True) config = NPURunConfig(memory_config=mem_config, session_config=session_config) |