Skip to content

Commit

Permalink
changing choices for argument backend from "pytorch" to "torch" in in…
Browse files Browse the repository at this point in the history
…stall_keras documentation, adding "torch" to backends vector
  • Loading branch information
16EAGLE committed May 3, 2024
1 parent 80af555 commit 8c1ad68
Show file tree
Hide file tree
Showing 2 changed files with 6 additions and 6 deletions.
6 changes: 3 additions & 3 deletions R/install.R
Original file line number Diff line number Diff line change
Expand Up @@ -5,10 +5,10 @@
#' @param envname Name of or path to a Python virtual environment
#' @param extra_packages Additional Python packages to install alongside Keras
#' @param python_version Passed on to `reticulate::virtualenv_starter()`
#' @param backend Which backend(s) to install. Accepted values include `"tensorflow"`, `"jax"` and `"pytorch"`
#' @param backend Which backend(s) to install. Accepted values include `"tensorflow"`, `"jax"` and `"torch"`
#' @param gpu whether to install a GPU capable version of the backend.
#' @param restart_session Whether to restart the R session after installing (note this will only occur within RStudio).
#' @param ... reserved for future compatability.
#' @param ... reserved for future compatibility.
#'
#' @returns No return value, called for side effects.
#'
Expand All @@ -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"),
backend = c("tensorflow", "jax", "torch"),
# backend = "tf-nightly",
gpu = NA,
restart_session = TRUE) {
Expand Down
6 changes: 3 additions & 3 deletions man/install_keras.Rd

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

0 comments on commit 8c1ad68

Please sign in to comment.