fix multicores when using the torch model

e-sensing · Jul 29, 2024 · 790b947 · 790b947
1 parent 061a438
commit 790b947
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/R/sits_classify.R b/R/sits_classify.R
@@ -293,7 +293,9 @@ sits_classify.raster_cube <- function(data,
         proc_bloat = proc_bloat
     )
     # Update multicores parameter
-    if ("xgb_model" %in% .ml_class(ml_model) || .is_torch_model(ml_model))
+    if ("xgb_model" %in% .ml_class(ml_model))
+        multicores <- 1
+    else if (.torch_mps_enabled(ml_model) || .torch_cuda_enabled(ml_model))
         multicores <- 1
     else
         # Update multicores parameter