mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-06-26 07:22:12 +00:00
079bf3cb26
* [misc] update pre-commit * [misc] run pre-commit * [misc] remove useless configuration files * [misc] ignore cuda for clang-format
16 lines
342 B
Python
16 lines
342 B
Python
from .base import Actor, Critic, RewardModel
|
|
from .lora import LoRAModule, convert_to_lora_module
|
|
from .loss import LogExpLoss, LogSigLoss, PolicyLoss, ValueLoss
|
|
|
|
__all__ = [
|
|
"Actor",
|
|
"Critic",
|
|
"RewardModel",
|
|
"PolicyLoss",
|
|
"ValueLoss",
|
|
"LogSigLoss",
|
|
"LogExpLoss",
|
|
"LoRAModule",
|
|
"convert_to_lora_module",
|
|
]
|