protomotions.agents.peft.prior_amp_agent module

Discrete-prior PEFT RLFT agent with AMP discriminator rewards.

class protomotions.agents.peft.prior_amp_agent.DiscretePriorPEFTRLFTAMPAgent(fabric, env, config, root_dir=None)

Bases: AMPAgentMixin, DiscretePriorPEFTRLFTAgent

Discrete-prior PEFT RLFT with AMP-style reward shaping.

__init__(

fabric,

env,

config,

root_dir=None,

)