Keras: LSTM unit nums vs timesteps
The n_neurons (LSTM_nums) on the first input layer should match those of n_features or in your case data_dim.

Further reading: LSTM Neural Network for Time Series Prediction