github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-08-05 10:10:32 +00:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source 
* better logger using rich

* remove deepspeed in zero requirements
This commit is contained in:
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:

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