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