Struct SgdState

Source

pub struct SgdState {
    pub lr: f64,
    pub momentum: f64,
    pub velocity: Vec<f64>,
}

Expand description

SGD optimizer state.

Phase 2b: the velocity: Vec<f64> is the canonical per-parameter state buffer. The field stays pub for backward compatibility with existing readers (cjc-eval / cjc-mir-exec read state.velocity.len()), but new code should prefer the typed accessors velocity_at, set_velocity_at, n_params so that parameter-position vs node- index confusion is caught at compile time.

See ADR-0022 (Phase 2 — typed-ID newtypes) for the policy.

Fields§

§lr: f64§momentum: f64§velocity: Vec<f64>

Struct SgdState Copy item path

Fields§

Implementations§

impl SgdState

pub fn new(n_params: usize, lr: f64, momentum: f64) -> Self

pub fn n_params(&self) -> usize

pub fn velocity_at(&self, p: ParamIdx) -> f64

pub fn set_velocity_at(&mut self, p: ParamIdx, value: f64)

Auto Trait Implementations§

impl Freeze for SgdState

impl RefUnwindSafe for SgdState

impl Send for SgdState

impl Sync for SgdState

impl Unpin for SgdState

impl UnsafeUnpin for SgdState

impl UnwindSafe for SgdState

Blanket Implementations§

impl<T> Any for Twhere T: 'static + ?Sized,

fn type_id(&self) -> TypeId

impl<T> Borrow<T> for Twhere T: ?Sized,

fn borrow(&self) -> &T

impl<T> BorrowMut<T> for Twhere T: ?Sized,

fn borrow_mut(&mut self) -> &mut T

impl<T> From<T> for T

fn from(t: T) -> T

impl<T, U> Into<U> for Twhere U: From<T>,

fn into(self) -> U

impl<T> IntoEither for T

fn into_either(self, into_left: bool) -> Either<Self, Self>

fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>where F: FnOnce(&Self) -> bool,

impl<T> Pointable for T

const ALIGN: usize

type Init = T

unsafe fn init(init: <T as Pointable>::Init) -> usize

unsafe fn deref<'a>(ptr: usize) -> &'a T

unsafe fn deref_mut<'a>(ptr: usize) -> &'a mut T

unsafe fn drop(ptr: usize)

impl<T, U> TryFrom<U> for Twhere U: Into<T>,

type Error = Infallible

fn try_from(value: U) -> Result<T, <T as TryFrom<U>>::Error>

impl<T, U> TryInto<U> for Twhere U: TryFrom<T>,

type Error = <U as TryFrom<T>>::Error

fn try_into(self) -> Result<U, <U as TryFrom<T>>::Error>

Struct SgdState

impl<T> Any for T
where T: 'static + ?Sized,

impl<T> Borrow<T> for T
where T: ?Sized,

impl<T> BorrowMut<T> for T
where T: ?Sized,

impl<T, U> Into<U> for T
where U: From<T>,

fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
where F: FnOnce(&Self) -> bool,

impl<T, U> TryFrom<U> for T
where U: Into<T>,

impl<T, U> TryInto<U> for T
where U: TryFrom<T>,