v3 beta.6

GiviMAD · Sep 1, 2023 · 0bd2d5a · 0bd2d5a
1 parent a32bcc2
commit 0bd2d5a
Show file tree

Hide file tree

Showing 3 changed files with 25 additions and 22 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -1,14 +1,14 @@
 [package]
 name = "rustpotter-cli"
-version = "3.0.0-beta.5"
+version = "3.0.0-beta.6"
 edition = "2021"
 license = "Apache-2.0"
 description = "CLI for Rustpotter, an open source wakeword spotter forged in rust."
 authors = ["Miguel Álvarez Díez <miguelwork92@gmail.com>"]
 repository = "https://github.com/GiviMAD/rustpotter"
 exclude = ["tools/**",".github",".gitignore"]
 [dependencies]
-rustpotter = { version = "3.0.0-beta.5", features = ["debug", "audio", "record"] }
+rustpotter = { version = "3.0.0-beta.6", features = ["debug", "audio", "record"] }
 ctrlc = "3.2.2"
 clap = { version = "4.1.6", features = ["derive"] }
 hound = "3.4.0"

diff --git a/src/cli/train.rs b/src/cli/train.rs
@@ -23,6 +23,9 @@ pub struct TrainCommand {
     #[clap(short = 'e', long, default_value_t = 1000)]
     /// Number of backward and forward cycles to run
     epochs: usize,
+    #[clap(long, default_value_t = 1)]
+    /// Number of epochs for testing the model and print the progress.  
+    test_epochs: usize,
     #[clap(short = 'c', long, default_value_t = 16)]
     /// Number of extracted mel-frequency cepstral coefficients
     mfcc_size: u16,
@@ -43,6 +46,7 @@ pub fn train(command: TrainCommand) -> Result<(), String> {
         command.test_dir,
         command.learning_rate,
         command.epochs,
+        command.test_epochs,
         command.mfcc_size,
         model,
     )