pub struct ModelWeights {
pub layers: Vec<DenseLayer>,
}Expand description
A collection of named dense layers (binary-serialisable model weights).
Fields§
§layers: Vec<DenseLayer>Layers in order.
Implementations§
Source§impl ModelWeights
impl ModelWeights
Sourcepub fn add_layer(&mut self, layer: DenseLayer)
pub fn add_layer(&mut self, layer: DenseLayer)
Append a layer.
Sourcepub fn get_layer(&self, name: &str) -> Option<&DenseLayer>
pub fn get_layer(&self, name: &str) -> Option<&DenseLayer>
Look up a layer by name.
Sourcepub fn total_params(&self) -> usize
pub fn total_params(&self) -> usize
Total trainable parameter count.
Trait Implementations§
Source§impl Clone for ModelWeights
impl Clone for ModelWeights
Source§fn clone(&self) -> ModelWeights
fn clone(&self) -> ModelWeights
Returns a duplicate of the value. Read more
1.0.0 · Source§fn clone_from(&mut self, source: &Self)
fn clone_from(&mut self, source: &Self)
Performs copy-assignment from
source. Read moreSource§impl Debug for ModelWeights
impl Debug for ModelWeights
Source§impl Default for ModelWeights
impl Default for ModelWeights
Source§fn default() -> ModelWeights
fn default() -> ModelWeights
Returns the “default value” for a type. Read more
Auto Trait Implementations§
impl Freeze for ModelWeights
impl RefUnwindSafe for ModelWeights
impl Send for ModelWeights
impl Sync for ModelWeights
impl Unpin for ModelWeights
impl UnsafeUnpin for ModelWeights
impl UnwindSafe for ModelWeights
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
Source§impl<T> CloneToUninit for Twhere
T: Clone,
impl<T> CloneToUninit for Twhere
T: Clone,
Source§impl<SS, SP> SupersetOf<SS> for SPwhere
SS: SubsetOf<SP>,
impl<SS, SP> SupersetOf<SS> for SPwhere
SS: SubsetOf<SP>,
Source§fn to_subset(&self) -> Option<SS>
fn to_subset(&self) -> Option<SS>
The inverse inclusion map: attempts to construct
self from the equivalent element of its
superset. Read moreSource§fn is_in_subset(&self) -> bool
fn is_in_subset(&self) -> bool
Checks if
self is actually part of its subset T (and can be converted to it).Source§fn to_subset_unchecked(&self) -> SS
fn to_subset_unchecked(&self) -> SS
Use with care! Same as
self.to_subset but without any property checks. Always succeeds.Source§fn from_subset(element: &SS) -> SP
fn from_subset(element: &SS) -> SP
The inclusion map: converts
self to the equivalent element of its superset.