trinity.common.rewards.countdown_reward module

Base Reward Function Class.

class trinity.common.rewards.countdown_reward.CountDownRewardFn[source]

Bases: RewardFn

A reward function that rewards for countdown task. Ref: Jiayi-Pan/TinyZero verl/utils/reward_score/countdown.py

__init__()[source]