ColossalAI/applications/Chat/coati/trainer/callbacks/__init__.py

6 lines
193 B
Python

from .base import Callback
from .performance_evaluator import PerformanceEvaluator
from .save_checkpoint import SaveCheckpoint
__all__ = ['Callback', 'PerformanceEvaluator', 'SaveCheckpoint']