Skip to content

Commit 0baaf2f

Browse files
committed
Update total_training_steps to 300 in benchmark configurations
1 parent 25125e6 commit 0baaf2f

5 files changed

Lines changed: 5 additions & 5 deletions

File tree

tests/bench/benchmark_frozenlake/benchmark_frozenlake.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -65,7 +65,7 @@ ajet:
6565
save_freq: 99999
6666
test_freq: 99999
6767
total_epochs: 99999
68-
total_training_steps: 25
68+
total_training_steps: 300
6969
nnodes: 1
7070
n_gpus_per_node: 8
7171
logger: swanlab

tests/bench/benchmark_frozenlakelora/benchmark_frozenlakelora.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -65,7 +65,7 @@ ajet:
6565
save_freq: 99999
6666
test_freq: 99999
6767
total_epochs: 99999
68-
total_training_steps: 25
68+
total_training_steps: 300
6969
nnodes: 1
7070
n_gpus_per_node: 8
7171
logger: swanlab

tutorial/example_frozenlake/frozenlake_easy.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -64,7 +64,7 @@ ajet:
6464
save_freq: 99999
6565
test_freq: 99999
6666
total_epochs: 99999
67-
total_training_steps: 25
67+
total_training_steps: 300
6868
nnodes: 1
6969
n_gpus_per_node: 8
7070
logger: swanlab

tutorial/example_frozenlake/frozenlake_hard.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -63,7 +63,7 @@ ajet:
6363
save_freq: 99999
6464
test_freq: 99999
6565
total_epochs: 99999
66-
total_training_steps: 25
66+
total_training_steps: 300
6767
nnodes: 1
6868
n_gpus_per_node: 8
6969
logger: swanlab

tutorial/example_werewolves/werewolves.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -57,7 +57,7 @@ ajet:
5757
save_freq: 5
5858
test_freq: 99999
5959
total_epochs: 99999
60-
total_training_steps: 25
60+
total_training_steps: 300
6161
nnodes: 2
6262
n_gpus_per_node: 8
6363

0 commit comments

Comments
 (0)