From ef7046b2c84954e0c0aac083696e4bb9acbd95ee Mon Sep 17 00:00:00 2001 From: Maddox Werts Date: Wed, 19 Mar 2025 16:01:24 -0400 Subject: [PATCH] Fixed Inference spelling mistake --- project/src/config.rs | 4 ++-- project/src/main.rs | 4 ++-- project/src/neural.rs | 8 ++++---- project/src/neural/{infrence.rs => inference.rs} | 0 4 files changed, 8 insertions(+), 8 deletions(-) rename project/src/neural/{infrence.rs => inference.rs} (100%) diff --git a/project/src/config.rs b/project/src/config.rs index a3a9fe7..27ff69d 100644 --- a/project/src/config.rs +++ b/project/src/config.rs @@ -4,7 +4,7 @@ use std::env; // Enums pub enum OperationMode { Training, - Infrence, + Inference, } // Functions @@ -15,7 +15,7 @@ pub fn get_operation_mode() -> Option { // Getting operation mode match args[1].as_str() { "training" => Some(OperationMode::Training), - "infrence" => Some(OperationMode::Infrence), + "inference" => Some(OperationMode::Inference), _ => None, } } diff --git a/project/src/main.rs b/project/src/main.rs index 4e25152..1d190de 100644 --- a/project/src/main.rs +++ b/project/src/main.rs @@ -13,10 +13,10 @@ fn main() { // Creating a Neural Network with the Operation Mode match operation_mode { - None => panic!("Main: `OperationMode` not defined!"), Some(mode) => { neural = NeuralNetwork::new(mode); - } + }, + _ => panic!("Main: `OperationMode` not defined!"), } // Starting the network diff --git a/project/src/neural.rs b/project/src/neural.rs index 812baa1..069b0cf 100644 --- a/project/src/neural.rs +++ b/project/src/neural.rs @@ -1,6 +1,6 @@ // Libraries mod data; -mod infrence; +mod inference; mod model; mod training; use super::config::OperationMode; @@ -44,7 +44,7 @@ impl NeuralNetwork { ); // Infer the model - infrence::infer::( + inference::infer::( MODEL_DIRECTORY, device, burn::data::dataset::vision::MnistDataset::test() @@ -58,7 +58,7 @@ impl NeuralNetwork { let device = burn::backend::wgpu::WgpuDevice::default(); // Infer the model - infrence::infer::( + inference::infer::( MODEL_DIRECTORY, device, burn::data::dataset::vision::MnistDataset::test() @@ -71,7 +71,7 @@ impl NeuralNetwork { // Switching based on mode match self.mode { OperationMode::Training => self.train(), - OperationMode::Infrence => self.infer(), + OperationMode::Inference => self.infer(), } } } diff --git a/project/src/neural/infrence.rs b/project/src/neural/inference.rs similarity index 100% rename from project/src/neural/infrence.rs rename to project/src/neural/inference.rs