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

Update cifar10_dws_cnn.py

Browse Source
This commit is contained in:
Alexios Maras
2024-07-23 19:40:12 +03:00
committed by GitHub
parent 95e338b8a2
commit 697de4cbcf
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mpq/cifar10_dws_cnn/cifar10_dws_cnn.py
+2 -2
View File
@@ -29,8 +29,8 @@ X_test = (np.transpose(X_test, (0,3,1,2)))
X_val = (np.transpose(X_val, (0,3,1,2))) X_val = (np.transpose(X_val, (0,3,1,2)))
BATCH_SIZE = 128 BATCH_SIZE = 128
epochs = 1 epochs = [10, 10]
lr = 0.0001 lr = [0.001, 0.0001]
class DepthwiseBlock(nn.Module): class DepthwiseBlock(nn.Module):
def __init__(self, in_channels, out_channels): def __init__(self, in_channels, out_channels):