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

11 lines
244 B
Python
Raw Normal View History

from .base import OnPolicyTrainer, SLTrainer
2023-03-28 12:25:36 +00:00
from .ppo import PPOTrainer
from .rm import RewardModelTrainer
from .sft import SFTTrainer
__all__ = [
'SLTrainer', 'OnPolicyTrainer',
'RewardModelTrainer', 'SFTTrainer',
'PPOTrainer'
]