github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-25 03:31:56 +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
4 changed files with 4 additions and 3 deletions
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
from pathlib import Path
from typing import Union
from rich.logging import RichHandler
from colossalai.context.parallel_mode import ParallelMode
_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: