From f4981d0d031982ff29300e6226bc90e51a0f813c Mon Sep 17 00:00:00 2001
From: yongchao1 <297389370@qq.com>
Date: Wed, 11 Jan 2023 06:57:30 +0000
Subject: [PATCH] update
 TensorFlow/contrib/cv/TVNet_ID0951_for_TensorFlow/train_epe_sintel.py.

Signed-off-by: yongchao1 <297389370@qq.com>
---
 .../contrib/cv/TVNet_ID0951_for_TensorFlow/train_epe_sintel.py  | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/TensorFlow/contrib/cv/TVNet_ID0951_for_TensorFlow/train_epe_sintel.py b/TensorFlow/contrib/cv/TVNet_ID0951_for_TensorFlow/train_epe_sintel.py
index b025c3c63..b923e11d8 100644
--- a/TensorFlow/contrib/cv/TVNet_ID0951_for_TensorFlow/train_epe_sintel.py
+++ b/TensorFlow/contrib/cv/TVNet_ID0951_for_TensorFlow/train_epe_sintel.py
@@ -145,6 +145,8 @@ init = tf.global_variables_initializer()
 config = tf.ConfigProto()
 custom_op = config.graph_options.rewrite_options.custom_optimizers.add()
 custom_op.name = "NpuOptimizer"
+custom_op.parameter_map["graph_memory_max_size"].s = tf.compat.as_bytes(str(27 * 1024 * 1024 * 1024))
+custom_op.parameter_map["variable_memory_max_size"].s = tf.compat.as_bytes(str(4 * 1024 * 1024 * 1024))
 config.graph_options.rewrite_options.remapping = RewriterConfig.OFF  # 必须显式关闭
 config.graph_options.rewrite_options.memory_optimization = RewriterConfig.OFF  # 必须显式关闭
 sess = tf.Session(config=config)
-- 
Gitee