From 64c8a7b1b2263072c0520f339de34f2e9561eb84 Mon Sep 17 00:00:00 2001 From: bxdd Date: Tue, 21 Nov 2023 15:15:20 +0800 Subject: [PATCH] [MNT] add verbose in learnware.init --- learnware/__init__.py | 18 ++++++++++-------- 1 file changed, 10 insertions(+), 8 deletions(-) diff --git a/learnware/__init__.py b/learnware/__init__.py index 4813623..fabc31b 100644 --- a/learnware/__init__.py +++ b/learnware/__init__.py @@ -8,7 +8,7 @@ from .utils import is_torch_available, setup_seed logger = get_module_logger("Initialization") -def init(**kwargs): +def init(verbose=True, **kwargs): """Init learnware package Parameters @@ -24,14 +24,16 @@ def init(**kwargs): C.reset() - config_file = os.path.join(C.root_path, "config.json") - - if os.path.exists(config_file): - with open(config_file, "r") as f: - C.update(**dict(json.load(f))) + if verbose: + logger.info(f"init the learnware package with arguments {kwargs}") C.update(**{k: v for k, v in kwargs.items() if k in C}) - logger.info(f"init the learnware package with arguments {kwargs}") + config_file = os.path.join(C.root_path, "config.json") + if os.path.exists(config_file): + if verbose: + logger.info(f"init the learnware package with config file {config_file}") + with open(config_file, "r") as fin_config: + C.update(**dict(json.load(fin_config))) ## random seed deterministic = kwargs.get("deterministic", True) @@ -57,4 +59,4 @@ if not is_torch_available(verbose=False): ) # default init package -init() +init(verbose=False)