提交 4151d90c 编写于 作者: L liuwei1031 提交者: Zeng Jinle

improve the hint message of memory optimize, test=develop (#18260)

上级 ff399fd7
......@@ -12,7 +12,6 @@
# See the License for the specific language governing permissions and
# limitations under the License.
import logging
import multiprocessing
import os
import six
......@@ -215,7 +214,8 @@ class CompiledProgram(object):
self._build_strategy.enable_inplace = False
elif not self._build_strategy.memory_optimize or not self._build_strategy.enable_inplace:
# remind the user to try our memmory optimize strategy
logging.warn("""
six.print_(
"""
You can try our memory optimize feature to save your memory usage:
# create a build_strategy variable to set memory optimize option
build_strategy = compiler.BuildStrategy()
......@@ -236,7 +236,8 @@ class CompiledProgram(object):
# if you need to fetch conv1, then:
conv1.persistable = True
""")
""",
file=sys.stderr)
return self
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册