ColossalAI/applications/ChatGPT/chatgpt/trainer/__init__.py

6 lines
148 B
Python

from .base import Trainer
from .ppo import PPOTrainer
from .rm import RewardModelTrainer
__all__ = ['Trainer', 'PPOTrainer', 'RewardModelTrainer']