Monitoring Driving in a Monotonous Environment: Classification and Recognition of Driving Fatigue Based on Long Short-Term Memory Network

Table 5

Dimensions of each layer of the network.

Layer	Input size	Output size
Input	[64, 1, 2]	—
LSTM 1	[64, 1, 2]	[256, 2]
LSTM 2	[256, 2]	[256, 64]
LSTM output	—	[64, 1, 64]
Linear 1	[64, 64]	[64, 32]
Linear 2	[64, 32]	[64, 4]
Output	—	[64, 4]

Note. — indicates that the value was not assigned with any meaning for the related indicator.