I guess this view operation might remove the batch dimension and flatten all elements into the feature dim:
x = x.view(-1, 90000)

Replace it with:
x = x.view(x.size(0), -1)

to keep the batch size constant and adapt the number of input features for the following linear layer, if necessary.