revert previous and retry

rstudio · Feb 28, 2024 · 3000b97 · 3000b97
1 parent bf6bd24
commit 3000b97
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tests/testthat/test-luz.R b/tests/testthat/test-luz.R
@@ -10,12 +10,13 @@ y_train <- scaled_cars[1:25, 1, drop=FALSE]
 
 set.seed(1)
 
+gc() # to see memory available on CI
+
 luz_fit <- torch::nn_linear %>%
     luz::setup(loss = torch::nnf_mse_loss, optimizer = torch::optim_sgd) %>%
     luz::set_hparams(in_features = ncol(x_train), out_features = 1) %>%
     luz::set_opt_hparams(lr = 0.01) %>%
-    luz::fit(list(x_train, y_train), verbose = FALSE, dataloader_options = list(batch_size = 5),
-             accelerator = luz::accelerator(cpu = TRUE))
+    luz::fit(list(x_train, y_train), verbose = FALSE, dataloader_options = list(batch_size = 5))
 
 v <- vetiver_model(
     luz_fit,