Skip to content

Commit

Permalink
run examples on cpu
Browse files Browse the repository at this point in the history
  • Loading branch information
sebffischer committed Jan 26, 2024
1 parent 2f66af1 commit 285736c
Show file tree
Hide file tree
Showing 3 changed files with 3 additions and 3 deletions.
2 changes: 1 addition & 1 deletion man-roxygen/learner_example.R
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,7 @@
#' # Define the Learner and set parameter values
#' <%= sprintf("learner = lrn(\"%s\")", id)%>
#' learner$param_set$set_values(
#' <%= paste0(" ", paste0(c(param_vals, "batch_size = 1", "epochs = 1"), collapse = ", "))%>
#' <%= paste0(" ", paste0(c(param_vals, "batch_size = 1", "epochs = 1", device = "cpu"), collapse = ", "))%>
#' )
#'
#' # Define a Task
Expand Down
2 changes: 1 addition & 1 deletion man/mlr_learners.mlp.Rd

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

2 changes: 1 addition & 1 deletion man/mlr_learners.torch_featureless.Rd

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

0 comments on commit 285736c

Please sign in to comment.