We proceeded by changing the input size of the model.
The input size I applied is 1024x512.
And the following error occurred.
ValueError: Expected input batch_size (32706) to match target batch_size (32679)