Struct border_tch_agent::sac::Actor

pub struct Actor<P>where
    P: SubModel<Output = (Tensor, Tensor)>,
    P::Config: DeserializeOwned + Serialize + OutDim,{ /* private fields */ }

Represents a stochastic policy for SAC agents.

Implementations

impl<P> Actor<P>where P: SubModel<Output = (Tensor, Tensor)>, P::Config: DeserializeOwned + Serialize + OutDim,

pub fn build(config: ActorConfig<P::Config>, device: Device) -> Result<Actor<P>>

Constructs Actor.

pub fn forward(&self, x: &P::Input) -> (Tensor, Tensor)

Outputs the parameters of Gaussian distribution given an observation.

Trait Implementations

impl<P> Clone for Actor<P>where P: SubModel<Output = (Tensor, Tensor)>, P::Config: DeserializeOwned + Serialize + OutDim,

fn clone(&self) -> Self

Returns a copy of the value.

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

Performs copy-assignment from source.

impl<P> ModelBase for Actor<P>where P: SubModel<Output = (Tensor, Tensor)>, P::Config: DeserializeOwned + Serialize + OutDim,

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.