Skip to content

Commit 97888ba

Browse files
authored
Merge pull request deeplearning4j#957 from KonduitAI/master
Update examples from Dev Repository
2 parents ff3a274 + 115cb27 commit 97888ba

File tree

3 files changed

+9
-9
lines changed

3 files changed

+9
-9
lines changed

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

+3-3
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

+3-3
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
);

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

+3-3
Original file line numberDiff line numberDiff line change
@@ -46,9 +46,9 @@ public class Cartpole
4646
150000, //Max step
4747
150000, //Max size of experience replay
4848
32, //size of batches
49-
500, //target update (hard)
49+
100, //target update (hard)
5050
10, //num step noop warmup
51-
0.01, //reward scaling
51+
0.1, //reward scaling
5252
0.99, //gamma
5353
1.0, //td-error clipping
5454
0.1f, //min epsilon
@@ -58,7 +58,7 @@ public class Cartpole
5858

5959
public static DQNFactoryStdDense.Configuration CARTPOLE_NET =
6060
DQNFactoryStdDense.Configuration.builder()
61-
.l2(0.001).updater(new Adam(0.0005)).numHiddenNodes(16).numLayer(3).build();
61+
.updater(new Adam(0.001)).numHiddenNodes(16).numLayer(3).build();
6262

6363
public static void main(String[] args) throws IOException {
6464
cartPole();

0 commit comments

Comments
 (0)