diff --git a/man-roxygen/learner_example.R b/man-roxygen/learner_example.R index 2b7ab447..18637b8b 100644 --- a/man-roxygen/learner_example.R +++ b/man-roxygen/learner_example.R @@ -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 diff --git a/man/mlr_learners.mlp.Rd b/man/mlr_learners.mlp.Rd index 86c71288..9366ff0c 100644 --- a/man/mlr_learners.mlp.Rd +++ b/man/mlr_learners.mlp.Rd @@ -65,7 +65,7 @@ Is initialized to \code{0.5}. # Define the Learner and set parameter values learner = lrn("classif.mlp") learner$param_set$set_values( - layers = 1, d_hidden = 10, batch_size = 1, epochs = 1 + layers = 1, d_hidden = 10, batch_size = 1, epochs = 1, cpu ) # Define a Task diff --git a/man/mlr_learners.torch_featureless.Rd b/man/mlr_learners.torch_featureless.Rd index 9e4a602c..974503f5 100644 --- a/man/mlr_learners.torch_featureless.Rd +++ b/man/mlr_learners.torch_featureless.Rd @@ -54,7 +54,7 @@ Only those from \code{\link{LearnerTorch}}. # Define the Learner and set parameter values learner = lrn("classif.torch_featureless") learner$param_set$set_values( - batch_size = 1, epochs = 1 + batch_size = 1, epochs = 1, cpu ) # Define a Task