Method save

acme/agents/jax/bc/learning.py:151–152

(self)

    return [variables[name] for name in names]

  def save(self) -> TrainingState:
    return self._state

  def restore(self, state: TrainingState):
    self._state = state