File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed
rl4j-examples/src/main/java/org/deeplearning4j/examples/rl4j Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -46,9 +46,9 @@ public class Cartpole
46
46
150000 , //Max step
47
47
150000 , //Max size of experience replay
48
48
32 , //size of batches
49
- 500 , //target update (hard)
49
+ 100 , //target update (hard)
50
50
10 , //num step noop warmup
51
- 0.01 , //reward scaling
51
+ 0.1 , //reward scaling
52
52
0.99 , //gamma
53
53
1.0 , //td-error clipping
54
54
0.1f , //min epsilon
@@ -58,7 +58,7 @@ public class Cartpole
58
58
59
59
public static DQNFactoryStdDense .Configuration CARTPOLE_NET =
60
60
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 ();
62
62
63
63
public static void main (String [] args ) throws IOException {
64
64
cartPole ();
You can’t perform that action at this time.
0 commit comments