ChatGPT used Categorical CrossEntropy as the loss function. I asked it to use a different one. The updated code also works. Notice how it changed the loss function, but it also dropped the categorical transformation of labels. This is key for the new loss function work!