protomotions.agents.fine_tuning.pretrained_modules module

Mixin for agents that load frozen modules before model construction.

class protomotions.agents.fine_tuning.pretrained_modules.PretrainedModulesMixin

Bases: object

Load configured frozen modules through the BaseAgent setup lifecycle.

The mixin is intentionally independent of the training algorithm. PPO-style fine-tuning and supervised fine-tuning can both populate self.pretrained before create_model() and then drop that temporary handle after the model captures the modules it needs.

pretrained: Dict[str, <Mock object at 0x7fd694c52e10>[]]