Skip to content

Commit 06f8e9f

Browse files
committed
dummy cogames tweaks
1 parent 9f07669 commit 06f8e9f

File tree

2 files changed

+9
-7
lines changed

2 files changed

+9
-7
lines changed

metta

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
../metta

pufferlib/config/cogames.ini

Lines changed: 8 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -5,9 +5,10 @@ policy_name = Policy
55
rnn_name = Recurrent
66

77
[vec]
8-
num_envs = 64
9-
num_workers = 16
10-
batch_size = auto
8+
backend = Serial
9+
num_envs = 1
10+
num_workers = 1
11+
batch_size = 1
1112
zero_copy = True
1213

1314
[env]
@@ -16,7 +17,7 @@ render_mode = none
1617
variants = lonely_heart heart_chorus
1718

1819
[train]
19-
total_timesteps = 50_000_000
20-
batch_size = auto
21-
minibatch_size = 1024
22-
bptt_horizon = 64
20+
total_timesteps = 1_000_000
21+
batch_size = 64
22+
minibatch_size = 64
23+
bptt_horizon = 1

0 commit comments

Comments
 (0)