diff --git a/learnware/__init__.py b/learnware/__init__.py index 44b678f..b27c55e 100644 --- a/learnware/__init__.py +++ b/learnware/__init__.py @@ -36,12 +36,12 @@ def init(verbose=True, **kwargs): with open(config_file, "r") as fin_config: C.update(**dict(json.load(fin_config))) - ## random seed + # random seed deterministic = kwargs.get("deterministic", True) if deterministic: setup_seed(C.random_seed) - ## make dirs + # make dirs mkdir = kwargs.get("mkdir", True) if mkdir: os.makedirs(C.root_path, exist_ok=True) @@ -49,7 +49,7 @@ def init(verbose=True, **kwargs): os.makedirs(C.stdout_path, exist_ok=True) os.makedirs(C.cache_path, exist_ok=True) - ## ignore tensorflow warning + # ignore tensorflow warning tf_loglevel = kwargs.get("tf_loglevel", "2") os.environ["TF_CPP_MIN_LOG_LEVEL"] = tf_loglevel