Re-initializing neural network matrices on each set function.

nthnn · Apr 19, 2024 · 83d9ad5 · 83d9ad5
1 parent b9c8ee6
commit 83d9ad5
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/main.go b/main.go
@@ -90,6 +90,7 @@ func main() {
 
 		case N2CMU_SET_INPUT_COUNT:
 			network.InputCount = uart.ReadUint8()
+			network.InitMatrix()
 			break
 
 		case N2CMU_GET_INPUT_COUNT:
@@ -98,6 +99,7 @@ func main() {
 
 		case N2CMU_SET_HIDDEN_COUNT:
 			network.HiddenCount = uart.ReadUint8()
+			network.InitMatrix()
 			break
 
 		case N2CMU_GET_HIDDEN_COUNT:
@@ -106,6 +108,7 @@ func main() {
 
 		case N2CMU_SET_OUTPUT_COUNT:
 			network.OutputCount = uart.ReadUint8()
+			network.InitMatrix()
 			break
 
 		case N2CMU_GET_OUTPUT_COUNT: