Update README.md

mr-raccoon-97 · Oct 5, 2023 · 7e7fd0e · 7e7fd0e
1 parent 4c14144
commit 7e7fd0e
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/README.md b/README.md
@@ -47,14 +47,14 @@ struct Autoencoder : public net::Model<Autoencoder> {
     net::layer::Sequence encoder {
         net::layer::Linear(784, 128, net::initializer::He),
         net::layer::ReLU(),
-        net::layer::Linear(128, 64, net::initializer::He),
+        net::layer::Linear(128, 64), // default initializer He,
     };
 
     net::layer::Sequence decoder {
-        net::layer::Linear(64, 128, net::initializer::He),
+        net::layer::Linear(64, 128),
         net::layer::ReLU(),
-        net::layer::Linear(128, 784, net::initializer::He),
-        net::layer::LogSoftmax(1/*axis*/)
+        net::layer::Linear(128, 784, net::initializer::Xavier),
+        net::layer::LogSoftmax(1/*axis*/) 
     };
 
     net::Tensor forward(net::Tensor x) {