File tree Expand file tree Collapse file tree 3 files changed +9
-9
lines changed
rl4j-examples/src/main/java/org/deeplearning4j/examples/rl4j Expand file tree Collapse file tree 3 files changed +9
-9
lines changed Original file line number Diff line number Diff 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
Original file line number Diff line number Diff 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 );
Original file line number Diff line number Diff 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 ();
You can’t perform that action at this time.
0 commit comments