Struct border_tch_agent::dqn::DqnModel

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

Represents value functions for DQN agents.

Implementations

impl<Q> DqnModel<Q>where Q: SubModel<Output = Tensor>, Q::Config: DeserializeOwned + Serialize + OutDim,

pub fn build(config: DqnModelConfig<Q::Config>, device: Device) -> Self

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

Outputs the action-value given an observation.

Trait Implementations

impl<Q> Clone for DqnModel<Q>where Q: SubModel<Output = Tensor>, Q::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<Q> ModelBase for DqnModel<Q>where Q: SubModel<Output = Tensor>, Q::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.