Update cnn_model.h

2024-11-12 13:02:05 +08:00
parent 0566df80a2
commit 354f447ad1
1 changed files with 3 additions and 1 deletions
--- a/PORTING/CNN/cnn_model.h
+++ b/PORTING/CNN/cnn_model.h
@@ -38,7 +38,9 @@ typedef struct {
 #define FC1_WEIGHT_ARRSIZE (128*18432)		//2359296
 #define FC2_BIAS_ARRSIZE (7)
 #define FC2_WEIGHT_ARRSIZE (7*128)			//896
-#if 1
+
+#define is1250000 1
+#if is1250000
    #define DATA_ARRSIZE (1250000)
 #else
    #define DATA_ARRSIZE (100 * 100)