Struct Critic 

pub struct Critic<Q>
where
    Q: SubModel2<Output = Tensor>,
    Q::Config: DeserializeOwned + Serialize,
{ /* private fields */ }

Represents soft critic for SAC agents.

It takes observations and actions as inputs and outputs action values.

Implementations

impl<Q> Critic<Q>

where Q: SubModel2<Output = Tensor>, Q::Config: DeserializeOwned + Serialize,

pub fn build( config: CriticConfig<Q::Config>, device: Device, ) -> Result<Critic<Q>>

Constructs Critic.

pub fn forward(&self, obs: &Q::Input1, act: &Q::Input2) -> Tensor

Outputs the action-value given observations and actions.

Trait Implementations

impl<Q> Clone for Critic<Q>

where Q: SubModel2<Output = Tensor>, Q::Config: DeserializeOwned + Serialize,

fn clone(&self) -> Self

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl<Q> ModelBase for Critic<Q>

where Q: SubModel2<Output = Tensor>, Q::Config: DeserializeOwned + Serialize,

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

Trains the network given a loss.

fn get_var_store_mut(&mut self) -> &mut VarStore

Returns var_store as mutable reference.

fn get_var_store(&self) -> &VarStore

Returns var_store.

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

Save parameters of the neural network.

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

Load parameters of the neural network.