mirror of https://github.com/hpcaitech/ColossalAI
19 lines
675 B
Python
19 lines
675 B
Python
|
from .actor import Actor
|
||
|
|
from .bloom_actor import BLOOMActor
|
||
|
|
from .bloom_critic import BLOOMCritic
|
||
|
|
from .bloom_rm import BLOOMRM
|
||
|
|
from .critic import Critic
|
||
|
|
from .gpt_actor import GPTActor
|
||
|
|
from .gpt_critic import GPTCritic
|
||
|
|
from .gpt_rm import GPTRM
|
||
|
|
from .loss import PairWiseLoss, PolicyLoss, PPOPtxActorLoss, ValueLoss
|
||
|
|
from .opt_actor import OPTActor
|
||
|
|
from .opt_critic import OPTCritic
|
||
|
|
from .opt_rm import OPTRM
|
||
|
|
from .reward_model import RewardModel
|
||
|
|
|
||
|
|
__all__ = [
|
||
|
|
'Actor', 'Critic', 'RewardModel', 'PolicyLoss', 'ValueLoss', 'PPOPtxActorLoss', 'PairWiseLoss', 'GPTActor',
|
||
|
|
'GPTCritic', 'GPTRM', 'BLOOMActor', 'BLOOMCritic', 'BLOOMRM', 'OPTActor', 'OPTCritic', 'OPTRM'
|
||
|
|
]
|