evorl.ec.optimizers.utils

Module Contents

Classes

ExponentialScheduleSpec

Specification for an exponential schedule for HyperParam.

Functions

weight_sum

Weighted sum.

Data

API

class evorl.ec.optimizers.utils.ExponentialScheduleSpec[source]

Bases: evorl.types.PyTreeData

Specification for an exponential schedule for HyperParam.

decay: float

None

final: float

None

init: float

None

evorl.ec.optimizers.utils.optimizer_map

‘dict(…)’

evorl.ec.optimizers.utils.weight_sum(x: jax.Array, w: jax.Array) jax.Array[source]

Weighted sum.

Parameters:
  • x – (n, …)

  • w – (n,)