protomotions.agents.evaluators.mimic_evaluator module#

class protomotions.agents.evaluators.mimic_evaluator.MimicEpisodeContext(motion_ids, frame_limits)[source]#

Bases: object

Per-episode-batch state for mimic evaluation.

motion_ids: MockTensor#
frame_limits: MockTensor#
__init__(motion_ids, frame_limits)#
class protomotions.agents.evaluators.mimic_evaluator.MimicEvaluator(agent, fabric, config)[source]#

Bases: BaseEvaluator

Evaluator for Mimic agent’s motion tracking performance.

config: MimicEvaluatorConfig#
property motion_lib: MotionLib#

Motion library (from agent).

property motion_manager: MimicMotionManager#

Motion manager (from env).

initialize_eval()[source]#

Initialize evaluation tracking and cache the state eval will overwrite.

Saves the env state (the clip assignment is part of it) and the model’s rollout context, both restored by cleanup_after_evaluation.

run_evaluation()[source]#

Run evaluation across multiple motions.

process_eval_results()[source]#

Process results without changing training’s sampling distribution.

update_training_state(result)[source]#

Apply results when explicitly requested by the training caller.