Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
TillBeemelmanns committed Oct 13, 2022
1 parent b3c51cc commit 6540b07
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -169,9 +169,10 @@ Default parameters for the network architectures

### Model Zoo
| Model Architecture | Filename | Batch Size | Validation MAE | Evaluation SNNRMSE `iter=32` |
| --- |---------------------------------------|------------|----------------|------------------------------|
|--------------------|---------------------------------------|------------|----------------|------------------------------|
| Additive LSTM | `additive_lstm_32b_32iter.hdf5` | 32 | `2.6e-04` | `0.03473` |
| Additive LSTM Slim | `additive_lstm_128b_32iter_slim.hdf5` | 128 | `2.3e-04` | `0.03636` |
| Additive LSTM Demo | `additive_lstm_128b_32iter_demo.hdf5` | 128 | `2.9e-04` | `0.09762` |
| Oneshot LSTM | `oneshot_lstm_b128_32iter.hdf5` | 128 | `2.9e-04` | `0.05137` |
| Additive GRU | Will be uploaded soon | TBD | TBD | TBD |

Expand All @@ -189,6 +190,7 @@ After filling out the form, you will obtain a link where you can download the fo
- __`additive_lstm_32b_32iter.hdf5`__ (271 MB): Trained model with a bottleneck size of 32
- __`additive_lstm_128b_32iter_slim.hdf5`__ (67.9 MB): Trained model with a bottleneck size of 32
- __`oneshot_lstm_b128_32iter.hdf5`__ (68.1 MB): Trained model with a bottleneck size of 32
- __`additive_lstm_256b_32iter_demo.hdf5`__ (14.3 MB): Trained model with a bottleneck sizeof 32

## Inference & Evaluation Node
The inference and evaluation is conducted with ROS. The inferences and preprocessing nodes can be found
Expand Down

0 comments on commit 6540b07

Please sign in to comment.