protomotions.agents.evaluators.config module

Configuration classes for evaluators.

class protomotions.agents.evaluators.config.EvaluatorConfig(

_target_='protomotions.agents.evaluators.base_evaluator.BaseEvaluator',

eval_metrics_every=200,

)

Bases: object

Configuration for base evaluator.

Attributes:

eval_metrics_every: Evaluate metrics every N epochs. None = disabled.

eval_metrics_every: int | None = 200

__init__(

_target_='protomotions.agents.evaluators.base_evaluator.BaseEvaluator',

eval_metrics_every=200,

)

class protomotions.agents.evaluators.config.MotionWeightsRulesConfig(

motion_weights_update_success_discount=0.999,

motion_weights_update_failure_discount=0.999,

min_motion_weight='1/num_motions',

)

Bases: object

Configuration for motion weights update rule.

Attributes:

motion_weights_update_success_discount: Discount factor for successful motion weights. motion_weights_update_failure_discount: Discount for failed motions. 0 = set weight straight to 1. min_motion_weight: Minimum weight for any motion. ‘1/num_motions’ or float value.

motion_weights_update_success_discount: float = 0.999

motion_weights_update_failure_discount: float = 0.999

min_motion_weight: float | str = '1/num_motions'

__init__(

motion_weights_update_success_discount=0.999,

motion_weights_update_failure_discount=0.999,

min_motion_weight='1/num_motions',

)

class protomotions.agents.evaluators.config.MimicEvaluatorConfig(

_target_='protomotions.agents.evaluators.mimic_evaluator.MimicEvaluator',

eval_metrics_every=200,

eval_metric_keys=<factory>,

max_eval_steps=600,

save_predicted_motion_lib_every=3,

motion_weights_rules=<factory>,

)

Bases: EvaluatorConfig

Configuration for Mimic evaluator.

Attributes:

eval_metrics_every: Evaluate metrics every N epochs. None = disabled. eval_metric_keys: Metric keys to track during evaluation. max_eval_steps: Maximum steps per evaluation episode. save_predicted_motion_lib_every: Save pred_motion_lib every M evals. None = disabled. motion_weights_rules: Rules for updating motion sampling weights.

eval_metric_keys: List[str]

max_eval_steps: int = 600

save_predicted_motion_lib_every: int | None = 3

motion_weights_rules: MotionWeightsRulesConfig

__init__(

_target_='protomotions.agents.evaluators.mimic_evaluator.MimicEvaluator',

eval_metrics_every=200,

eval_metric_keys=<factory>,

max_eval_steps=600,

save_predicted_motion_lib_every=3,

motion_weights_rules=<factory>,

)