add gpu_memory parameter in sits_classify.sits functionn

e-sensing · Oct 25, 2023 · 4b51716 · 4b51716
1 parent 88b54ec
commit 4b51716
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/R/sits_classify.R b/R/sits_classify.R
@@ -165,6 +165,7 @@ sits_classify.sits <- function(data,
                                ...,
                                filter_fn = NULL,
                                multicores = 2L,
+                               gpu_memory = NULL,
                                progress = TRUE) {
     # Pre-conditions
     data <- .check_samples_ts(data)
@@ -181,6 +182,7 @@ sits_classify.sits <- function(data,
         ml_model = ml_model,
         filter_fn = filter_fn,
         multicores = multicores,
+        gpu_memory = gpu_memory,
         progress = progress
     )
     return(classified_ts)

diff --git a/man/sits_classify.Rd b/man/sits_classify.Rd