[D] Training model on tabular data resulting in high loss

Hello, I am training a model on tabular data which has already been preprocessed (scaled, PCA). Currently there are over 50k rows and 10 columns. The loss are high, not sure where I'm doing wrong

For context, I'm using MSE as my loss function, 0.01 learning rate and 256 batch size.

Thank you so much.

This is how my init code looks like:

class NN(nn.Module):
def init(self): super(NN, self).init()
# Tabular data processing layers
self.fc1 = nn.Linear(10, 64)
self.fc2 = …

context data function hello loss machinelearning pca rate tabular tabular data training