Update examples/torch/torch_gru/torch_gru.cpp

Co-authored-by: jatinchowdhury18 <[email protected]>
jatinchowdhury18 · Jul 30, 2023 · 153a3d3 · 153a3d3
1 parent 1aee26b
commit 153a3d3
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/examples/torch/torch_gru/torch_gru.cpp b/examples/torch/torch_gru/torch_gru.cpp
@@ -34,9 +34,9 @@ std::string getOutputFile(fs::path path)
     return path.string();
 }
 
-const int gru_input_size = 1;
-const int gru_hidden_size = 8;
-const int n_output_neurons = 1;
+constexpr int gru_input_size = 1;
+constexpr int gru_hidden_size = 8;
+constexpr int n_output_neurons = 1;
 
 using ModelType = RTNeural::ModelT<float, gru_input_size, n_output_neurons,
     RTNeural::GRULayerT<float, gru_input_size, gru_hidden_size>,