Update REA, REINFORCE, RANDOM, and BOHB

2020-07-14 11:53:21 +00:00
parent 168b08d9e6
commit 2c861f33c4
8 changed files with 79 additions and 88 deletions
--- a/exps/algos-v2/reinforce.py
+++ b/exps/algos-v2/reinforce.py
@@ -150,7 +150,7 @@ def main(xargs, api):
    start_time = time.time()
    log_prob, action = select_action( policy )
    arch   = policy.generate_arch( action )
-    reward, _, _, current_total_cost = api.simulate_train_eval(arch, xargs.dataset, '12')
+    reward, _, _, current_total_cost = api.simulate_train_eval(arch, xargs.dataset, hp='12')
    trace.append((reward, arch))
    total_costs.append(current_total_cost)