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 b025c3c63df26205a3b29f7e9eb0b2e73d584536..b923e11d895a05023ce61e721ea16f240cb6d38a 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)