trinity.common.rewards.reward_fn module
Base Reward Function Class.
-
class trinity.common.rewards.reward_fn.RewardFn(**kwargs)[source]
Bases: ABC
Base Reward Function Class.
-
abstract __init__(**kwargs) → None[source]
-
class trinity.common.rewards.reward_fn.RMGalleryFn(reward_name, **kwargs)[source]
Bases: RewardFn
Reward Function from RMGallery.
modelscope/RM-Gallery
-
__init__(reward_name, **kwargs)[source]