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)