Struct border_tch_agent::sac::EntCoef

pub struct EntCoef { /* private fields */ }

The entropy coefficient of SAC.

Implementations

impl EntCoef

pub fn new(mode: EntCoefMode, device: Device) -> Self

Constructs an instance of EntCoef.

pub fn alpha(&self) -> Tensor

Returns the entropy coefficient.

pub fn backward_step(&mut self, loss: &Tensor)

Does an optimization step given a loss.

pub fn update(&mut self, logp: &Tensor)

Update the parameter given an action probability vector.

pub fn save<T: AsRef<Path>>(&self, path: T) -> Result<()>

Save the parameter into a file.

pub fn load<T: AsRef<Path>>(&mut self, path: T) -> Result<()>

Save the parameter from a file.