Explicit type of the optimizer

fszewczyk · Nov 7, 2023 · 7f2207f · 7f2207f
1 parent d3ea8b8
commit 7f2207f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/main.cpp b/tests/main.cpp
@@ -14,7 +14,7 @@ int main() {
                 .build();
     // clang-format on
 
-    auto optimizer = Optimizer32(mlp->parameters(), 0.1);
+    auto optimizer = Optimizer<Type::float32>(mlp->parameters(), 0.1);
     auto lossFunction = Loss::MSE<Type::float32>;
 
     for (size_t epoch = 0; epoch < 1000; epoch++) {