Skip to content

Commit 2782df4

Browse files
committed
Set/get command for output weights via UART.
1 parent 62581c0 commit 2782df4

File tree

1 file changed

+14
-0
lines changed

1 file changed

+14
-0
lines changed

main.go

+14
Original file line numberDiff line numberDiff line change
@@ -171,6 +171,20 @@ func main() {
171171
}
172172
break
173173

174+
case N2CMU_SET_OUTPUT_WEIGHTS:
175+
for i := 0; i < len(network.OutputWeights); i++ {
176+
network.OutputWeights[i] = uart.ReadFloat32()
177+
}
178+
179+
uart.WriteOk()
180+
break
181+
182+
case N2CMU_GET_OUTPUT_WEIGHTS:
183+
for i := 0; i < len(network.OutputWeights); i++ {
184+
uart.WriteFloat32(network.OutputWeights[i])
185+
}
186+
break
187+
174188
case N2CMU_SET_EPOCH_COUNT:
175189
epoch = uart.ReadUint16()
176190
break

0 commit comments

Comments
 (0)