more state 2

CMakeLists.txt

@@ -1,5 +1,5 @@
 cmake_minimum_required(VERSION 3.25)
-project(COSC-4P80-Final-Project VERSION 0.0.16)
+project(COSC-4P80-Final-Project VERSION 0.0.17)
 
 option(ENABLE_ADDRSAN "Enable the address sanitizer" OFF)
 option(ENABLE_UBSAN "Enable the ub sanitizer" OFF)

src/MNIST.cpp

@@ -617,7 +617,8 @@ namespace fp
             std::exit(-1);
         }
 
-        state << i;
+        // user can skip this if required.
+        state << i - 1;
         state << '\n';
         state << last_epoch_save;
         state << '\n';