pub struct ConvolutionalModel { /* private fields */ }Expand description
Convolutional neural network implementation
Implementations§
Source§impl ConvolutionalModel
impl ConvolutionalModel
Sourcepub fn new(config: NeuralSpatialConfig, device: Device) -> Result<Self>
pub fn new(config: NeuralSpatialConfig, device: Device) -> Result<Self>
Create a new convolutional neural network model
Trait Implementations§
Source§impl NeuralModel for ConvolutionalModel
impl NeuralModel for ConvolutionalModel
Source§fn forward(&self, input: &NeuralInputFeatures) -> Result<NeuralSpatialOutput>
fn forward(&self, input: &NeuralInputFeatures) -> Result<NeuralSpatialOutput>
Forward pass through the model
Source§fn config(&self) -> &NeuralSpatialConfig
fn config(&self) -> &NeuralSpatialConfig
Get model configuration
Source§fn update_parameters(&mut self, params: &HashMap<String, Tensor>) -> Result<()>
fn update_parameters(&mut self, params: &HashMap<String, Tensor>) -> Result<()>
Update model parameters
Source§fn metrics(&self) -> NeuralPerformanceMetrics
fn metrics(&self) -> NeuralPerformanceMetrics
Get model performance metrics
Source§fn memory_usage(&self) -> usize
fn memory_usage(&self) -> usize
Get memory usage in bytes
Auto Trait Implementations§
impl Freeze for ConvolutionalModel
impl !RefUnwindSafe for ConvolutionalModel
impl Send for ConvolutionalModel
impl Sync for ConvolutionalModel
impl Unpin for ConvolutionalModel
impl UnsafeUnpin for ConvolutionalModel
impl !UnwindSafe for ConvolutionalModel
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> Instrument for T
impl<T> Instrument for T
Source§fn instrument(self, span: Span) -> Instrumented<Self>
fn instrument(self, span: Span) -> Instrumented<Self>
Source§fn in_current_span(self) -> Instrumented<Self>
fn in_current_span(self) -> Instrumented<Self>
Source§impl<T> IntoEither for T
impl<T> IntoEither for T
Source§fn into_either(self, into_left: bool) -> Either<Self, Self>
fn into_either(self, into_left: bool) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left is true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read moreSource§fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left(&self) returns true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read more