Skip to content

GitLab

Unverified Commit 969c4c2f authored by liuzhe-lz's avatar liuzhe-lz Committed by GitHub
Browse files

disable logging in nnictl (#3268)

parent 58820396
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
nni/tools/nnictl/nnictl.py
View file @ 969c4c2f
...@@ -2,6 +2,7 @@ ...@@ -2,6 +2,7 @@
# Licensed under the MIT license. # Licensed under the MIT license.
import argparse import argparse
import logging
import os import os
import pkg_resources import pkg_resources
from colorama import init from colorama import init
...@@ -32,6 +33,8 @@ def nni_info(*args): ...@@ -32,6 +33,8 @@ def nni_info(*args):
print('please run "nnictl {positional argument} --help" to see nnictl guidance') print('please run "nnictl {positional argument} --help" to see nnictl guidance')
def parse_args(): def parse_args():
logging.getLogger().setLevel(logging.ERROR)
'''Definite the arguments users need to follow and input''' '''Definite the arguments users need to follow and input'''
parser = argparse.ArgumentParser(prog='nnictl', description='use nnictl command to control nni experiments') parser = argparse.ArgumentParser(prog='nnictl', description='use nnictl command to control nni experiments')
parser.add_argument('--version', '-v', action='store_true') parser.add_argument('--version', '-v', action='store_true')
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment