#[repr(C)]pub struct AdamParams {
pub lr: f32,
pub beta1: f32,
pub beta2: f32,
pub eps: f32,
pub weight_decay: f32,
pub t: f32,
pub _pad: [f32; 2],
}Expand description
Hyper-parameters for ComputeEngine::adamw_update; laid out to match
the WGSL uniform.
Fields§
§lr: f32Learning rate.
beta1: f32First-moment decay.
beta2: f32Second-moment decay.
eps: f32Denominator epsilon.
weight_decay: f32Decoupled weight decay.
t: f321-based step number, used for bias correction.
_pad: [f32; 2]Padding to a 16-byte multiple; set to zero.
Trait Implementations§
Auto Trait Implementations§
impl Freeze for AdamParams
impl RefUnwindSafe for AdamParams
impl Send for AdamParams
impl Sync for AdamParams
impl Unpin for AdamParams
impl UnsafeUnpin for AdamParams
impl UnwindSafe for AdamParams
Blanket Implementations§
impl<T> AnyBitPattern for Twhere
T: Pod,
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> CheckedBitPattern for Twhere
T: AnyBitPattern,
impl<T> CheckedBitPattern for Twhere
T: AnyBitPattern,
Source§type Bits = T
type Bits = T
Self must have the same layout as the specified Bits except for
the possible invalid bit patterns being checked during
is_valid_bit_pattern.Source§fn is_valid_bit_pattern(_bits: &T) -> bool
fn is_valid_bit_pattern(_bits: &T) -> bool
If this function returns true, then it must be valid to reinterpret
bits
as &Self.