schihei/mixed-precision-neural-networks-on-risc-v-cores
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update cmsis_cnn.py

Browse Source
This commit is contained in:
Alexios Maras
2024-07-23 19:40:41 +03:00
committed by GitHub
parent 697de4cbcf
commit 48ca62244f
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mpq/cmsis_cnn/cmsis_cnn.py
+2 -2
View File
@@ -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 X_val = (np.transpose(X_val, (0,3,1,2)) - 128.0)/255.0
BATCH_SIZE = 32 BATCH_SIZE = 32
epochs = 1 epochs = [50, 25]
lr = 0.0001 lr = [0.0001, 0.00001]
class CMSIS_CNN(nn.Module): class CMSIS_CNN(nn.Module):
def __init__(self): def __init__(self):