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

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

diff --git a/mpq/cifar10_dws_cnn/cifar10_dws_cnn.py b/mpq/cifar10_dws_cnn/cifar10_dws_cnn.py
index 9cced02..e2d5bc2 100644
--- a/mpq/cifar10_dws_cnn/cifar10_dws_cnn.py
+++ b/mpq/cifar10_dws_cnn/cifar10_dws_cnn.py
@@ -29,8 +29,8 @@ X_test = (np.transpose(X_test, (0,3,1,2)))
 X_val = (np.transpose(X_val, (0,3,1,2)))
 
 BATCH_SIZE = 128
-epochs = 1
-lr = 0.0001
+epochs = [10, 10]
+lr = [0.001, 0.0001]
 
 class DepthwiseBlock(nn.Module):
     def __init__(self, in_channels, out_channels):