File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 1+ [base]
2+ env_name = admiral
3+ checkpoint_interval = 8
4+ seed = 42
5+
6+ [vec]
7+ total_agents = 4096
8+ num_buffers = 1
9+ num_threads = 4
10+ num_policies = 2
11+ hist_policy_percent = 0.1
12+ hist_policy_hidden_size = 128
13+ hist_policy_num_layers = 2
14+
15+ [selfplay]
16+ enabled = 1
17+ max_size = 100
18+ seed = 42
19+ opp_timeout_steps = 16_777_216
20+ eval_games = 4096
21+ eval_pool_size = 8
22+
23+ [env]
24+ num_agents = 2 # Must match N_TEAMS; also used by the shared evaluator.
25+ curriculum_level = 1
26+ width = 2000
27+ height = 2000
28+ reward_damage_mult = 0.5
29+ reward_kill = 0.19897151
30+ penalty_hit_ally = -0.542492926
31+ penalty_used_volley = 0
32+ penalty_stationary = -0.00454199314
33+
34+ [policy]
35+ hidden_size = 128
36+ num_layers = 2
37+
38+ [train]
39+ gpus = 1
40+ total_timesteps = 536_870_912
41+ learning_rate = 0.000396882882
42+ anneal_lr = 0
43+ gamma = 0.995000005
44+ gae_lambda = 0.979191363
45+ replay_ratio = 1.95889282
46+ clip_coef = 0.169899464
47+ vf_coef = 3
48+ vf_clip_coef = 0.100000001
49+ max_grad_norm = 1.38442755
50+ ent_coef = 3.39512771e-05
51+ anneal_ent_coef = 0
52+ momentum = 0.997428834
53+ minibatch_size = 8192
54+ horizon = 128
55+
56+ [sweep]
57+ # Self-play pool evaluation replaces this training metric as Protein's objective.
58+ metric = perf
59+ metric_distribution = linear
60+ goal = maximize
61+ max_runs = 1000
62+ gpus = 1
63+ downsample = 32
64+ prune_pareto = True
65+
66+ [sweep.train.learning_rate]
67+ distribution = log_normal
68+ min = 0.0003
69+ max = 0.003
70+ scale = 0.5
71+
72+ [sweep.train.ent_coef]
73+ distribution = log_normal
74+ min = 0.00001
75+ max = 0.001
76+ scale = auto
You can’t perform that action at this time.
0 commit comments