Number of features = 272
And it’s one hot encoded at that
And u have data of of Length = 5 / batch
Assuming this tabular data is of 5 columns,
The u are having a data of (5, 5, 272) per batch.
This is really really large.
U should try to use a different vectorizing method other than the one hot encoding method simply because the number of feature is way way much and can lead to higher computational time.