Bases: LMRewardsOperationalMetric
Prompt tokens served from cache during reward computation.
Example:
program.compile(
metrics=[
synalinks.metrics.RewardCachedTokens(),
],
)
Source code in synalinks/src/metrics/lm_metrics.py
| @synalinks_export("synalinks.metrics.RewardCachedTokens")
class RewardCachedTokens(LMRewardsOperationalMetric):
"""Prompt tokens served from cache during reward computation.
Example:
```python
program.compile(
metrics=[
synalinks.metrics.RewardCachedTokens(),
],
)
```
"""
def __init__(self, name="reward_cached_tokens"):
super().__init__(name=name)
def result(self):
return int(self._delta("cached_tokens"))
|