Browse Source

[update] add default args for logger method

tags/v0.4.10
Yunfan Shao GitHub 5 years ago
parent
commit
0de2ec8823
No known key found for this signature in database GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 2 deletions
  1. +4
    -2
      fastNLP/io/_logger.py

+ 4
- 2
fastNLP/io/_logger.py View File

@@ -133,10 +133,12 @@ def _get_logger(name=None, level='INFO'):


class FastNLPLogger(logging.Logger):
def add_file(self, path, level):
def add_file(self, path='./log.txt', level='INFO'):
"""add log output file and level"""
_add_file_handler(self, path, level)

def set_stdout(self, stdout, level):
def set_stdout(self, stdout='tqdm', level='INFO'):
"""set stdout format and level"""
_set_stdout_handler(self, stdout, level)

_logger = _init_logger(path=None)


Loading…
Cancel
Save