From c6105e224ed4b0dbeefa6af41d8a82fd4b5f5178 Mon Sep 17 00:00:00 2001 From: Sang Choe Date: Tue, 10 Oct 2023 10:15:57 -0400 Subject: [PATCH] remove device in Config (deprecation) --- examples/logistic_regression_hpo/test.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/logistic_regression_hpo/test.py b/examples/logistic_regression_hpo/test.py index ef65ca6..dcaeb9f 100644 --- a/examples/logistic_regression_hpo/test.py +++ b/examples/logistic_regression_hpo/test.py @@ -109,8 +109,8 @@ def on_inner_loop_start(self): child_config = Config(type="cg", cg_iterations=3, cg_alpha=0.1, unroll_steps=100) engine_config = EngineConfig(train_iters=5000) -parent = Parent(name="outer", config=parent_config, device=device) -child = Child(name="inner", config=child_config, device=device) +parent = Parent(name="outer", config=parent_config) +child = Child(name="inner", config=child_config) problems = [parent, child] u2l = {parent: [child]}