evorl.algorithms.contrib.td3_ep

Module Contents

Classes

TD3WorkflowMod

TD3Workflow with total_episode termination condition.

API

class evorl.algorithms.contrib.td3_ep.TD3WorkflowMod(env: evorl.envs.Env, agent: evorl.agent.Agent, optimizer: optax.GradientTransformation, evaluator: evorl.evaluators.Evaluator, replay_buffer: evorl.replay_buffers.AbstractReplayBuffer, config: omegaconf.DictConfig)

Bases: evorl.algorithms.td3.TD3Workflow

TD3Workflow with total_episode termination condition.

learn(state: evorl.types.State) → evorl.types.State