Struct AdamWOptimizer 

pub struct AdamWOptimizer { /* private fields */ }

AdamW optimizer (Adam with decoupled weight decay)

Implementations

impl AdamWOptimizer

pub fn instance_of<V: NapiRaw>(env: Env, value: V) -> Result<bool>

impl AdamWOptimizer

pub fn into_reference( val: AdamWOptimizer, env: Env, ) -> Result<Reference<AdamWOptimizer>>

pub fn into_instance(self, env: Env) -> Result<ClassInstance<AdamWOptimizer>>

impl AdamWOptimizer

pub fn new(param_count: u32, learning_rate: f64, weight_decay: f64) -> Self

Create a new AdamW optimizer

Arguments

• param_count - Number of parameters
• learning_rate - Learning rate
• weight_decay - Weight decay coefficient

pub fn with_betas( param_count: u32, learning_rate: f64, weight_decay: f64, beta1: f64, beta2: f64, ) -> Self

Create with custom betas

pub fn step( &mut self, params: Float32Array, gradients: Float32Array, ) -> Float32Array

Perform an optimization step

Returns

Updated parameter array

pub fn reset(&mut self)

Reset optimizer state

pub fn learning_rate(&self) -> f64

Get current learning rate

pub fn set_learning_rate(&mut self, lr: f64)

Set learning rate

pub fn weight_decay(&self) -> f64

Get weight decay

Trait Implementations

impl FromNapiMutRef for AdamWOptimizer

unsafe fn from_napi_mut_ref( env: napi_env, napi_val: napi_value, ) -> Result<&'static mut Self>

Safety

impl FromNapiRef for AdamWOptimizer

unsafe fn from_napi_ref( env: napi_env, napi_val: napi_value, ) -> Result<&'static Self>

Safety

impl FromNapiValue for &AdamWOptimizer

unsafe fn from_napi_value(env: napi_env, napi_val: napi_value) -> Result<Self>

Safety

fn from_unknown(value: JsUnknown) -> Result<Self, Error>

impl FromNapiValue for &mut AdamWOptimizer

unsafe fn from_napi_value(env: napi_env, napi_val: napi_value) -> Result<Self>

Safety

fn from_unknown(value: JsUnknown) -> Result<Self, Error>

impl ObjectFinalize for AdamWOptimizer

fn finalize(self, env: Env) -> Result<(), Error>

impl ToNapiValue for AdamWOptimizer

unsafe fn to_napi_value( env: napi_env, val: AdamWOptimizer, ) -> Result<napi_value>

Safety

impl TypeName for &AdamWOptimizer

fn type_name() -> &'static str

fn value_type() -> ValueType

impl TypeName for &mut AdamWOptimizer

fn type_name() -> &'static str

fn value_type() -> ValueType

impl TypeName for AdamWOptimizer

fn type_name() -> &'static str

fn value_type() -> ValueType

impl ValidateNapiValue for &AdamWOptimizer

unsafe fn validate(env: napi_env, napi_val: napi_value) -> Result<napi_value>

Safety

impl ValidateNapiValue for &mut AdamWOptimizer

unsafe fn validate(env: napi_env, napi_val: napi_value) -> Result<napi_value>

Safety