Skip to content

Commit b65b94d

Browse files
author
Tim-phant
committed
fix sweep , breakout config for sweeps
1 parent acd7d44 commit b65b94d

2 files changed

Lines changed: 31 additions & 29 deletions

File tree

config/boxoban.ini

Lines changed: 29 additions & 28 deletions
Original file line numberDiff line numberDiff line change
@@ -2,9 +2,9 @@
22
env_name = boxoban
33

44
[vec]
5-
total_agents = 32768
6-
num_buffers = 4
7-
num_threads = 1
5+
total_agents = 4096
6+
num_buffers = 8
7+
num_threads = 8
88

99
[env]
1010
num_agents = 1
@@ -14,31 +14,32 @@ target_loss_pen_coeff = 0
1414
max_steps = 150
1515

1616
[policy]
17-
hidden_size = 512
18-
num_layers = 3.32422
19-
expansion_factor = 1
17+
num_layers = 2
18+
hidden_size = 64
2019

2120
[train]
2221
gpus = 1
23-
seed = 42
24-
total_timesteps = 880580001
25-
learning_rate = 0.00134234
26-
anneal_lr = 1
27-
min_lr_ratio = 0.37872
28-
gamma = 0.989717
29-
gae_lambda = 0.759273
30-
replay_ratio = 1.6234
31-
clip_coef = 0.01
32-
vf_coef = 5
33-
vf_clip_coef = 5
34-
max_grad_norm = 1.20325
35-
ent_coef = 0.000188411
36-
beta1 = 0.995526
37-
beta2 = 0.999536
38-
eps = 1e-14
39-
minibatch_size = 32768
40-
horizon = 64
41-
vtrace_rho_clip = 3.13347
42-
vtrace_c_clip = 2.75328
43-
prio_alpha = 0.453827
44-
prio_beta0 = 0.765589
22+
total_timesteps = 94_000_000
23+
beta1 = 0.7279714073125252
24+
beta2 = 0.9986265112492152
25+
clip_coef = 0.6746497927896418
26+
ent_coef = 0.0033240721522812535
27+
eps = 0.00008339460257113628
28+
gae_lambda = 0.948721675814334
29+
gamma = 0.9721246598992744
30+
learning_rate = 0.1
31+
max_grad_norm = 1.8109182724544075
32+
minibatch_size = 65_536
33+
prio_alpha = 0.1
34+
prio_beta0 = 0.8247156461060179
35+
replay_ratio = 1.4242098997083206
36+
vf_clip_coef = 1.2291681640124468
37+
vf_coef = 1.2195502588297364
38+
vtrace_c_clip = 1.0830442742115065
39+
vtrace_rho_clip = 2.1017317041552603
40+
41+
[sweep]
42+
gpus = 1
43+
max_suggestion_cost = 10800
44+
45+

pufferlib/sweep.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -146,7 +146,8 @@ def _params_from_puffer_sweep(sweep_config, only_include=None):
146146

147147
for name, param in sweep_config.items():
148148
if name in ('method', 'metric', 'metric_distribution', 'goal', 'downsample', 'use_gpu', 'prune_pareto',
149-
'sweep_only', 'max_suggestion_cost', 'early_stop_quantile', 'gpus', 'max_runs'):
149+
'sweep_only', 'max_suggestion_cost', 'early_stop_quantile', 'gpus', 'max_runs', 'match_enemy_model_path', 'match_num_games',
150+
'match_enemy_hidden_size', 'match_enemy_num_layers'):
150151
continue
151152

152153
assert isinstance(param, dict), f'Param {name} is not a dict'

0 commit comments

Comments
 (0)