github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

[log] better logging display with rich (#426) 

* better logger using rich

* remove deepspeed in zero requirements
pull/427/head
Jiarui Fang 2022-03-16 09:51:15 +08:00 committed by GitHub
parent 3f70a2b12f
commit 54229cd33e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
colossalai/logging/logging.py
View File

@ -5,11 +5,12 @@ import colossalai
import logging import logging
from pathlib import Path from pathlib import Path
from typing import Union from typing import Union
from rich.logging import RichHandler
from colossalai.context.parallel_mode import ParallelMode from colossalai.context.parallel_mode import ParallelMode
_FORMAT = 'colossalai - %(name)s - %(asctime)s %(levelname)s: %(message)s' _FORMAT = 'colossalai - %(name)s - %(asctime)s %(levelname)s: %(message)s'
logging.basicConfig(level=logging.INFO, format=_FORMAT) logging.basicConfig(level=logging.INFO, format=_FORMAT, handlers=[RichHandler()])
class DistributedLogger: class DistributedLogger:

2
requirements/requirements-test.txt
View File

@ -1,3 +1,3 @@
pytest pytest
rpyc rpyc
matplotlib matplotlib

1
requirements/requirements-zero.txt
View File

@ -1 +0,0 @@
deepspeed

1
requirements/requirements.txt
View File

@ -6,3 +6,4 @@ psutil
tensorboard tensorboard
packaging packaging
pre-commit pre-commit
rich