CPM-9G-8B/9G-Train/cpm/utils/__init__.py

8 lines
222 B
Python
Raw Normal View History

2024-02-27 14:33:33 +08:00
from .config import Config
from .data_utils import pad
from .data_utils import pad_raw
from .gradient_shrink import gradient_shrink
from .log import logger
from .log import LogManager
from .object import allgather_objects