Skip to content

Commit 0fdf6fc

Browse files
authored
Merge pull request #4 from Bam4d/async_hyperparams_2
better hyperparameters for async models
2 parents ff3a274 + 34f3cf8 commit 0fdf6fc

File tree

2 files changed

+6
-6
lines changed

2 files changed

+6
-6
lines changed

rl4j-examples/src/main/java/org/deeplearning4j/examples/rl4j/A3CCartpole.java

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -42,11 +42,11 @@ public class A3CCartpole {
4242
200, //Max step By epoch
4343
500000, //Max step
4444
16, //Number of threads
45-
5, //t_max
45+
10, //t_max
4646
10, //num step noop warmup
47-
0.01, //reward scaling
47+
0.1, //reward scaling
4848
0.99, //gamma
49-
10.0 //td-error clipping
49+
1.0 //td-error clipping
5050
);
5151

5252

rl4j-examples/src/main/java/org/deeplearning4j/examples/rl4j/AsyncNStepCartpole.java

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -41,12 +41,12 @@ public class AsyncNStepCartpole {
4141
200, //Max step By epoch
4242
300000, //Max step
4343
16, //Number of threads
44-
5, //t_max
44+
10, //t_max
4545
100, //target update (hard)
4646
10, //num step noop warmup
47-
0.01, //reward scaling
47+
0.1, //reward scaling
4848
0.99, //gamma
49-
100.0, //td-error clipping
49+
1.0, //td-error clipping
5050
0.1f, //min epsilon
5151
9000 //num step for eps greedy anneal
5252
);

0 commit comments

Comments
 (0)