Exclude "torch" from default backends

rstudio · May 3, 2024 · 6a01221 · 6a01221
1 parent 8c1ad68
commit 6a01221
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/R/install.R b/R/install.R
@@ -19,7 +19,7 @@ install_keras <- function(
     extra_packages = c("scipy", "pandas", "Pillow", "pydot", "ipython", "tensorflow_datasets"),
     python_version = ">=3.9,<=3.11",
     # backend = "tensorflow",
-    backend = c("tensorflow", "jax", "torch"),
+    backend = c("tensorflow", "jax"),
     # backend = "tf-nightly",
     gpu = NA,
     restart_session = TRUE) {