xpag.agents.flax_agents.sdqn.sdqn.TrainingState#

class TrainingState(critic_up_optimizer_state, critic_up_params, target_critic_up_params, critic_low_optimizer_state, critic_low_params, target_critic_low_params, key, steps)#

Bases: object

Contains training state for the learner.

Methods

replace

"Returns a new object replacing the specified fields with new values.

Attributes

`critic_up_optimizer_state`
`critic_up_params`
`target_critic_up_params`
`critic_low_optimizer_state`
`critic_low_params`
`target_critic_low_params`
`key`
`steps`

replace(**updates)#: “Returns a new object replacing the specified fields with new values.