Struct GpuSgd 

pub struct GpuSgd { /* private fields */ }

GPU stochastic gradient descent with optional momentum / Nesterov.

Implementations

impl GpuSgd

pub fn new(params: SgdParams) -> Result<Self, GpuOptimError>

Create the optimizer with the default WebGPU backend.

pub fn with_config( params: SgdParams, config: GpuOptimizerConfig, ) -> Result<Self, GpuOptimError>

Create the optimizer on an explicit backend.

pub fn backend(&self) -> GpuBackend

Backend actually in use.

pub fn move_to_gpu(&mut self) -> Result<(), GpuOptimError>

Upload the optimizer state to device memory.

Inherent alias for GpuOptimizer::move_to_gpu so callers do not need a turbofish to pin the unused dimension parameter.

pub fn move_to_cpu(&mut self) -> Result<(), GpuOptimError>

Download the optimizer state back to host memory.

Inherent alias for GpuOptimizer::move_to_cpu; see Self::move_to_gpu.

pub fn to_gpu(&mut self) -> Result<(), GpuOptimError>

👎Deprecated since 0.3.2:

renamed to move_to_gpu

Deprecated alias for Self::move_to_gpu.

pub fn to_cpu(&mut self) -> Result<(), GpuOptimError>

👎Deprecated since 0.3.2:

renamed to move_to_cpu

Deprecated alias for Self::move_to_cpu.

pub fn is_gpu_available(&self) -> bool

Whether a real device (not the CPU fallback) is backing this optimizer.

pub fn step_count(&self) -> u64

Number of updates applied so far.

Trait Implementations

impl<D: Dimension> GpuOptimizer<f32, D> for GpuSgd

fn is_gpu_available(&self) -> bool

Check if GPU acceleration is available

fn move_to_gpu(&mut self) -> Result<(), GpuOptimError>

Move optimizer state to GPU

fn move_to_cpu(&mut self) -> Result<(), GpuOptimError>

Move optimizer state back to CPU

fn step_gpu( &mut self, params: &mut Array<f32, D>, gradients: &Array<f32, D>, ) -> Result<(), GpuOptimError>

Perform optimization step on GPU

fn to_gpu(&mut self) -> Result<(), GpuOptimError>

👎Deprecated since 0.3.2:

renamed to move_to_gpu

Deprecated alias for Self::move_to_gpu.

fn to_cpu(&mut self) -> Result<(), GpuOptimError>

👎Deprecated since 0.3.2:

renamed to move_to_cpu

Deprecated alias for Self::move_to_cpu.