Skip to content

Commit

Permalink
Update highmodel.html
Browse files Browse the repository at this point in the history
  • Loading branch information
kai9987kai authored Feb 1, 2025
1 parent 9a5b1bd commit 7e42b43
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions highmodel.html
Original file line number Diff line number Diff line change
Expand Up @@ -351,7 +351,7 @@
buildCommanderDQN() {
const model = tf.sequential();
// Very large model: hidden layers of 32768, 16384, 8192, 4096, 2048, 1024 units.
model.add(tf.layers.dense({ units: 22768, inputShape: [9] }));
model.add(tf.layers.dense({ units: 22760, inputShape: [9] }));
model.add(tf.layers.batchNormalization());
model.add(tf.layers.leakyReLU({ alpha: 0.1 }));
model.add(tf.layers.dropout({ rate: 0.2 }));
Expand All @@ -363,11 +363,11 @@
model.add(tf.layers.batchNormalization());
model.add(tf.layers.leakyReLU({ alpha: 0.1 }));
model.add(tf.layers.dropout({ rate: 0.2 }));
model.add(tf.layers.dense({ units: 4093 }));
model.add(tf.layers.dense({ units: 4090 }));
model.add(tf.layers.batchNormalization());
model.add(tf.layers.leakyReLU({ alpha: 0.1 }));
model.add(tf.layers.dropout({ rate: 0.2 }));
model.add(tf.layers.dense({ units: 2042 }));
model.add(tf.layers.dense({ units: 2040 }));
model.add(tf.layers.batchNormalization());
model.add(tf.layers.leakyReLU({ alpha: 0.1 }));
model.add(tf.layers.dense({ units: 1024, activation: 'linear' }));
Expand Down

0 comments on commit 7e42b43

Please sign in to comment.