Skip to main content
. 2023 Feb 23;12:e83035. doi: 10.7554/eLife.83035

Table 1. Parameter and operation counts for various networks.

The number of neurons in the input, hidden, and output layers are d, n, and N, respectively. Note these counts do not include parameters of the readout layer, since said layer contributes the same number of parameters for each network (N(n+1) and Nn with and without a bias) and are for fixed initial states.

Network Trainable Parameters State update operations
2-layer fully connected n(d+1) O(nd2)
MPN n(d+1)+2 O(nd2)
MPNpre n(d+1)+2 O(nd2)
Vanilla RNN n(n+d+1) O(n3)+O(nd2)
GRU 3n(n+d+1) O(3n3)+O(3nd2)
LSTM 4n(n+d+1) O(4n3)+O(4nd2)