pub struct DuelingNetwork { /* private fields */ }Expand description
Generic neural network with dueling architecture Can be used by any RL algorithm (DQN, PPO, SAC, etc.)
Implementations§
Source§impl DuelingNetwork
impl DuelingNetwork
Sourcepub fn new(config: NetworkConfig, vb: VarBuilder<'_>) -> CandleResult<Self>
pub fn new(config: NetworkConfig, vb: VarBuilder<'_>) -> CandleResult<Self>
Create network from configuration
Sourcepub fn forward(
&self,
state: &Tensor,
training: bool,
) -> CandleResult<(Tensor, Tensor, Tensor)>
pub fn forward( &self, state: &Tensor, training: bool, ) -> CandleResult<(Tensor, Tensor, Tensor)>
Forward pass through network
Sourcepub fn get_config(&self) -> &NetworkConfig
pub fn get_config(&self) -> &NetworkConfig
Get network configuration
Trait Implementations§
Auto Trait Implementations§
impl Freeze for DuelingNetwork
impl !RefUnwindSafe for DuelingNetwork
impl Send for DuelingNetwork
impl Sync for DuelingNetwork
impl Unpin for DuelingNetwork
impl UnsafeUnpin for DuelingNetwork
impl !UnwindSafe for DuelingNetwork
Blanket Implementations§
Source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
Source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more
Source§impl<T> Instrument for T
impl<T> Instrument for T
Source§fn instrument(self, span: Span) -> Instrumented<Self>
fn instrument(self, span: Span) -> Instrumented<Self>
Source§fn in_current_span(self) -> Instrumented<Self>
fn in_current_span(self) -> Instrumented<Self>
Source§impl<T> IntoEither for T
impl<T> IntoEither for T
Source§fn into_either(self, into_left: bool) -> Either<Self, Self>
fn into_either(self, into_left: bool) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left is true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read moreSource§fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left(&self) returns true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read more