`nemo_gym.profile`#

Module Contents#

Classes#

`ProfileConfig`
`RewardProfilingMetrics`

Functions#

API#

class nemo_gym.profile.ProfileConfig(/, **data: typing.Any)[source]#

Bases: nemo_gym.config_types.BaseNeMoGymCLIConfig

input_jsonl_fpath: str#: ‘Field(…)’

rollouts_jsonl_fpath: str#: ‘Field(…)’

output_jsonl_fpath: str#: ‘Field(…)’

pass_threshold: Optional[float]#: ‘Field(…)’

class nemo_gym.profile.RewardProfilingMetrics(/, **data: typing.Any)[source]#

Bases: pydantic.BaseModel

avg_reward: float#: ‘Field(…)’

std_reward: float#: ‘Field(…)’

min_reward: float#: ‘Field(…)’

max_reward: float#: ‘Field(…)’

total_samples: int#: ‘Field(…)’

pass_rate: Optional[float]#: ‘Field(…)’

pass_rate_total: Optional[int]#: ‘Field(…)’

pass_rate_passed: Optional[int]#: ‘Field(…)’

pass_threshold: Optional[float]#: ‘Field(…)’

nemo_gym.profile.profile()[source]#