disable the gradient history recording & set model in evaluation (inf…

…erence) mode
bnosac · Mar 18, 2024 · 859c261 · 859c261
1 parent 057f779
commit 859c261
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 3 deletions.
diff --git a/DESCRIPTION b/DESCRIPTION
@@ -1,7 +1,7 @@
 Package: audio.vadsilero
 Type: Package
 Title: Voice Activity Detection using 'Silero'
-Version: 0.1
+Version: 0.2
 Maintainer: Jan Wijffels <[email protected]>
 Authors@R: c(
     person('Jan', 'Wijffels', role = c('aut', 'cre', 'cph'), email = '[email protected]', comment = "R wrapper"), 

diff --git a/NEWS.md b/NEWS.md
@@ -1,3 +1,7 @@
+## CHANGES IN audio.vadsilero VERSION 0.2
+
+- disable the gradient history recording & set model in evaluation (inference) mode
+
 ## CHANGES IN audio.vadsilero VERSION 0.1
 
 - Added function Silero to detect voice in audio

diff --git a/R/pkg.R b/R/pkg.R
@@ -1,3 +1,3 @@
 #' @importFrom wav read_wav
-#' @importFrom torch torch_tensor jit_load jit_scalar
+#' @importFrom torch torch_tensor jit_load jit_scalar autograd_set_grad_mode
 NULL
diff --git a/R/silero.R b/R/silero.R
@@ -93,7 +93,9 @@ silero <- function(file,
 }
 
 SILERO <- function(){
-    mod <- torch::jit_load(system.file(package = "audio.vadsilero", "models", "silero_vad.jit"))
+    autograd_set_grad_mode(FALSE)
+    mod <- jit_load(system.file(package = "audio.vadsilero", "models", "silero_vad.jit"))
+    mod$eval()
     out <- list(model = mod, version = "v4")
     class(out) <- "SILERO"
     out