From 52d1de6cec8b3388f439c204abcdf540d367cfb0 Mon Sep 17 00:00:00 2001 From: liuzhe Date: Thu, 21 Nov 2019 10:00:56 +0800 Subject: [PATCH 1/2] Print INFO log in "standalone" mode --- src/sdk/pynni/nni/platform/__init__.py | 1 + src/sdk/pynni/nni/platform/standalone.py | 6 ++++++ 2 files changed, 7 insertions(+) diff --git a/src/sdk/pynni/nni/platform/__init__.py b/src/sdk/pynni/nni/platform/__init__.py index cee80956e5..a2bce2f0e8 100644 --- a/src/sdk/pynni/nni/platform/__init__.py +++ b/src/sdk/pynni/nni/platform/__init__.py @@ -23,6 +23,7 @@ if trial_env_vars.NNI_PLATFORM is None: from .standalone import * + _init() elif trial_env_vars.NNI_PLATFORM == 'unittest': from .test import * elif trial_env_vars.NNI_PLATFORM in ('local', 'remote', 'pai', 'kubeflow', 'frameworkcontroller'): diff --git a/src/sdk/pynni/nni/platform/standalone.py b/src/sdk/pynni/nni/platform/standalone.py index 07da01d986..59f3b4d51d 100644 --- a/src/sdk/pynni/nni/platform/standalone.py +++ b/src/sdk/pynni/nni/platform/standalone.py @@ -19,6 +19,7 @@ # ================================================================================================== +import logging import json_tricks @@ -40,3 +41,8 @@ def send_metric(string): print('Final result:', metric['value']) elif metric['type'] == 'PERIODICAL': print('Intermediate result:', metric['value']) + +def _init(): + # print INFO log to stdout + logging.basicConfig() + logging.getLogger('nni').setLevel(logging.INFO) From 0f9fe4b89e0f67a823d4b961e8a3d87362748bdb Mon Sep 17 00:00:00 2001 From: liuzhe Date: Thu, 21 Nov 2019 10:15:21 +0800 Subject: [PATCH 2/2] init on importing --- src/sdk/pynni/nni/platform/__init__.py | 1 - src/sdk/pynni/nni/platform/standalone.py | 10 +++++----- 2 files changed, 5 insertions(+), 6 deletions(-) diff --git a/src/sdk/pynni/nni/platform/__init__.py b/src/sdk/pynni/nni/platform/__init__.py index a2bce2f0e8..cee80956e5 100644 --- a/src/sdk/pynni/nni/platform/__init__.py +++ b/src/sdk/pynni/nni/platform/__init__.py @@ -23,7 +23,6 @@ if trial_env_vars.NNI_PLATFORM is None: from .standalone import * - _init() elif trial_env_vars.NNI_PLATFORM == 'unittest': from .test import * elif trial_env_vars.NNI_PLATFORM in ('local', 'remote', 'pai', 'kubeflow', 'frameworkcontroller'): diff --git a/src/sdk/pynni/nni/platform/standalone.py b/src/sdk/pynni/nni/platform/standalone.py index 59f3b4d51d..7f752786b7 100644 --- a/src/sdk/pynni/nni/platform/standalone.py +++ b/src/sdk/pynni/nni/platform/standalone.py @@ -23,6 +23,11 @@ import json_tricks +# print INFO log to stdout +logging.basicConfig() +logging.getLogger('nni').setLevel(logging.INFO) + + def get_next_parameter(): pass @@ -41,8 +46,3 @@ def send_metric(string): print('Final result:', metric['value']) elif metric['type'] == 'PERIODICAL': print('Intermediate result:', metric['value']) - -def _init(): - # print INFO log to stdout - logging.basicConfig() - logging.getLogger('nni').setLevel(logging.INFO)