while comparing with the results provided in https://keras.io/examples/timeseries/timeseries_classification_transformer/, the dense layer is different. ```python dense_2 (Dense) (None, 128) 256 ['global_average_pooling1d_2[0 ][0]'] dropout_25 (Dropout) (None, 128) 0 ['dense_2[0][0]'] ``` how to fix it?