pub struct GANOptimizer { /* private fields */ }Expand description
GAN optimizer with stability improvements
Implementations§
Source§impl GANOptimizer
impl GANOptimizer
pub fn new( g_lr: f32, d_lr: f32, spectral_norm: bool, gradient_penalty_weight: f32, ) -> Self
pub fn step_discriminator( &mut self, d_params: &mut [Tensor], d_grads: &[Tensor], ) -> Result<()>
pub fn step_generator( &mut self, g_params: &mut [Tensor], g_grads: &[Tensor], ) -> Result<()>
Auto Trait Implementations§
impl Freeze for GANOptimizer
impl RefUnwindSafe for GANOptimizer
impl Send for GANOptimizer
impl Sync for GANOptimizer
impl Unpin for GANOptimizer
impl UnsafeUnpin for GANOptimizer
impl UnwindSafe for GANOptimizer
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
impl<ST, DT> CastableFrom<ST, Initialized, Initialized> for DT
impl<ST, DT> CastableFrom<ST, Uninit, Uninit> for DT
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