fix: some adjustments 🐛

Joker2770 · Aug 22, 2024 · 73b730f · 73b730f
1 parent 8f1e002
commit 73b730f
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/test/get_best_action_from_prob_test.cpp b/test/get_best_action_from_prob_test.cpp
@@ -39,9 +39,9 @@ int main(int argc, char *argv[])
   std::shared_ptr<NeuralNetwork> module = nullptr;
   if (argc <= 1)
   {
-    // cout << "Do not load weights. AI color = BLACK." << endl;
+    // std::cout << "Do not load weights. AI color = BLACK." << std::endl;
 
-    cout << "Warning: Find No weight path and color, assume they are mymodel and 1 (AI color:Black)" << endl;
+    std::cout << "Warning: Find No weight path and color, assume they are mymodel and 1 (AI color:Black)" << std::endl;
 #ifdef _WIN32
     module = std::make_shared<NeuralNetwork>("E:/Projects/AlphaZero-Onnx/python/mymodel.onnx", NUM_MCT_SIMS);
 #else
@@ -50,7 +50,7 @@ int main(int argc, char *argv[])
   }
   else
   {
-    cout << "Load weights: " << argv[1] << endl;
+    std::cout << "Load weights: " << argv[1] << std::endl;
     // wchar_t wchar[128] = {0};
     // swprintf(wchar,128,L"%S",argv[1]);