Merge pull request #1025 from OldLipe/feat/temp_seg

Fix cuda clean cache bug

R/api_classify.R

@@ -112,7 +112,9 @@
         # Apply the classification model to values
         values <- ml_model(values)
         # Clean the torch cache
-        torch::cuda_empty_cache()
+        if (torch::cuda_is_available()) {
+            torch::cuda_empty_cache()
+        }
         # Are the results consistent with the data input?
         .check_processed_values(values, input_pixels)
         # Log

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)