| commit | 6926092c1188f15fc4e75f236b13000ebcf97050 | [log] [tgz] |
|---|---|---|
| author | Tamas Hubai <efabless@htamas.net> | Mon Dec 05 22:53:55 2022 +0100 |
| committer | Tamas Hubai <efabless@htamas.net> | Mon Dec 05 22:53:55 2022 +0100 |
| tree | e7660e818dbd7a0373586053d2f4ef6ca9b4c1f7 | |
| parent | 90259320c6b9c014976c3cc7dc93ef4999267e3f [diff] |
Harden trainable_nn
Implements a simple neural network that supports on-chip training in addition to inference. The two hidden layers use leaky ReLU as their activation function while the output layer uses a rough approximation of softmax.
Unlike usual neural network implementations we use fixed-point saturation arithmetic and each neuron and synapse is represented as an individual instance. Inputs, outputs and weights, as well as forward and backward propagation can be managed through the wishbone bus.