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. https://github.com/modelscope/RM-Gallery

__init__(reward_name, **kwargs)[source]