Struct DqnModel 

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

Action value function model for DQN.

The architecture of the model is defined by the type parameter Q, which should implement SubModel. This takes SubModel::Input as input and outputs a tensor. The output tensor should have the same dimension as the number of actions.

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 observation(s).

pub fn param_stats(&self) -> Record

Trait Implementations

impl<Q> Clone for DqnModel<Q>

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

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 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.