minor changes

1 files changed, 10 insertions, 1 deletions

diff --git a/main.c b/main.c
index 7a1f62e..df976ea 100644
--- a/main.c
+++ b/main.c
@@ -8,7 +8,7 @@
 // #define DEBUG
 #define CHECK_SEED 69
 #define BIAS       20
-#define PASSES     999999
+#define PASSES     99999
 
 #define MIN(x, y) (((x) < (y)) ? (x) : (y))
 
@@ -111,6 +111,13 @@ int main()
 #endif
     }
 
+    for(size_t i = 0; i < PASSES; i++) {
+        train();
+#ifdef DEBUG
+        sleep(5);
+#endif
+    }
+
     puts("[INFO] Training completed");
     printf("[INFO] Operation took %ld seconds\n", time(0) - cur_time);
 
@@ -129,7 +136,9 @@ int main()
     printf("[INFO] Model success rate is %f%%\n", ((float)correct/PASSES)*100);
     printf("[INFO] Operation took %ld seconds\n", time(0) - cur_time);
 
+#ifdef DEBUG
     print_weight();
+#endif
 
     return 0;
 }