From 48ca62244fc6195e53a64f2447b1b8c33e033b1b Mon Sep 17 00:00:00 2001
From: Alexios Maras <100875700+alexmr09@users.noreply.github.com>
Date: Tue, 23 Jul 2024 19:40:41 +0300
Subject: [PATCH] Update cmsis_cnn.py

---
 mpq/cmsis_cnn/cmsis_cnn.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/mpq/cmsis_cnn/cmsis_cnn.py b/mpq/cmsis_cnn/cmsis_cnn.py
index 8435bf5..46ce4ac 100644
--- a/mpq/cmsis_cnn/cmsis_cnn.py
+++ b/mpq/cmsis_cnn/cmsis_cnn.py
@@ -28,8 +28,8 @@ X_test = (np.transpose(X_test, (0,3,1,2)) - 128.0)/255.0
 X_val = (np.transpose(X_val, (0,3,1,2)) - 128.0)/255.0
 
 BATCH_SIZE = 32
-epochs = 1
-lr = 0.0001
+epochs = [50, 25]
+lr = [0.0001, 0.00001]
 
 class CMSIS_CNN(nn.Module):
     def __init__(self):