Skip to content

Commit ba2c208

Browse files
committed
Set/get command for hidden gradient via UART.
1 parent 688bf3c commit ba2c208

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
@@ -213,6 +213,20 @@ func main() {
213213
}
214214
break
215215

216+
case N2CMU_SET_HIDDEN_GRAD:
217+
for i := 0; i < int(network.HiddenCount); i++ {
218+
network.HiddenGrad[i] = uart.ReadFloat32()
219+
}
220+
221+
uart.WriteOk()
222+
break
223+
224+
case N2CMU_GET_HIDDEN_GRAD:
225+
for i := 0; i < int(network.HiddenCount); i++ {
226+
uart.WriteFloat32(network.HiddenGrad[i])
227+
}
228+
break
229+
216230
case N2CMU_SET_EPOCH_COUNT:
217231
epoch = uart.ReadUint16()
218232
break

0 commit comments

Comments
 (0)