mirror of https://github.com/hpcaitech/ColossalAI
7 lines
161 B
Python
7 lines
161 B
Python
|
from .actor import Actor
|
||
|
|
from .critic import Critic
|
||
|
|
from .lm import LM
|
||
|
|
from .reward_model import RewardModel
|
||
|
|
|
||
|
|
__all__ = ['Actor', 'Critic', 'RewardModel', 'LM']
|