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

revert-18229-add_multi_gpu_install_check
liuwei1031 6 years ago committed by Zeng Jinle
parent ff399fd720
commit 4151d90c16

@ -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

Loading…
Cancel
Save