Skip to content

Commit 0073dcc

Browse files
committed
should work, test this on another machine
1 parent a489897 commit 0073dcc

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

benchmarks/rf_use_case/run_benchmark.R

+4-2
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,8 @@ library(here)
1212

1313
options(mlr3oml.cache = here("benchmarks", "data", "oml"))
1414

15+
# when working on the GPU server, don't forget to activate the mamba environment with the torch installation
16+
1517
# define the tasks
1618
cc18_small = fread(here(getOption("mlr3oml.cache"), "collections", "cc18_small.csv"))
1719

@@ -71,10 +73,10 @@ at = auto_tuner(
7173
tuner = tnr_mbo,
7274
resampling = rsmp("cv"),
7375
measure = msr("classif.acc"),
74-
term_evals = 10
76+
term_evals = 1000
7577
)
7678

77-
future::plan("multisession", workers = 64)
79+
future::plan("multisession", workers = 8)
7880

7981
lrn_rf = lrn("classif.ranger")
8082
design = benchmark_grid(

0 commit comments

Comments
 (0)