As stated before, the size of x is [10799, 53]. This is how I instantiate the model class: teacher_model = TeacherUpdated() This is how the start of my for loop looks that passes the x into the model:
for step_count, (x, y_gt) in enumerate(train_loader): #(x, y_gt)
        # Initialize gradients with 0
        optimizer.zero_grad()

        x= pd.DataFrame(data=x)
        x= torch.tensor(x.values)
        x = x.to(device)
        y_gt = y_gt.to(device)

        # Predict
        x = torch.flatten(x,start_dim=1, end_dim=-1)
        y_pred = teacher_model(x)