From c902e420173ae2c7f15c8b75ffc07c582d6555a7 Mon Sep 17 00:00:00 2001
From: Brett Laptop <tri11paragon@tpgc.me>
Date: Wed, 8 Jan 2025 17:19:51 -0500
Subject: [PATCH] third updaate

---
 CMakeLists.txt | 2 +-
 src/MNIST.cpp  | 4 ++++
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/CMakeLists.txt b/CMakeLists.txt
index d2cd27c..43b1b91 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -1,5 +1,5 @@
 cmake_minimum_required(VERSION 3.25)
-project(COSC-4P80-Final-Project VERSION 0.0.8)
+project(COSC-4P80-Final-Project VERSION 0.0.9)
 
 option(ENABLE_ADDRSAN "Enable the address sanitizer" OFF)
 option(ENABLE_UBSAN "Enable the ub sanitizer" OFF)
diff --git a/src/MNIST.cpp b/src/MNIST.cpp
index 0d0da2f..ddff1b8 100644
--- a/src/MNIST.cpp
+++ b/src/MNIST.cpp
@@ -508,6 +508,7 @@ namespace fp
 
         for (blt::i32 i = 0; i < runs; i++)
         {
+            BLT_TRACE("Starting run %d", i);
             auto local_ident = ident + std::to_string(i);
             NetworkType network{};
             if (restore)
@@ -542,6 +543,7 @@ namespace fp
                                max_pool<2, 2, 2, 2, relu<con<16, 5, 5, 1, 1,
                                                              max_pool<2, 2, 2, 2, relu<con<6, 5, 5, 1, 1,
                                                                                            input<matrix<blt::u8>>>>>>>>>>>>>>;
+        BLT_TRACE("Running deep learning tests");
         return run_network_tests<net_type_dl>(path, "deep_learning", runs, restore);
     }
 
@@ -555,6 +557,7 @@ namespace fp
                        relu<fc<120,
                                input<matrix<blt::u8>>>>>>>>;
 
+        BLT_TRACE("Running feed forward tests");
         return run_network_tests<net_type_ff>(path, "feed_forward", runs, restore);
     }
 
@@ -566,6 +569,7 @@ namespace fp
         python_dual_stacked_graph_program = binary_directory + "../graph.py";
         BLT_TRACE(binary_directory);
         BLT_TRACE(python_dual_stacked_graph_program);
+        BLT_TRACE("Running with batch size %d", batch_size);
 
         using namespace dlib;