xpag.agents.flax_agents.sac.sac_from_jaxrl.Model

class Model(step, apply_fn, params, tx, opt_state=None)

Bases: object

Methods

apply_gradient	rtype: Union[Tuple[Model, Any], Model]
create	rtype: Model
load	rtype: Model
replace	"Returns a new object replacing the specified fields with new values.
save

Attributes

opt_state
step
apply_fn
params
tx

__call__(*args, **kwargs)

Call self as a function.

replace(**updates)

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