Group Abstract Group Abstract

Message Boards Message Boards

Problem with CrossEntropyLossLayer["Index"] in NetTrain[]

I would appreciate feedback regarding an issue I am having with the CrossEntropyLossLayer["Index"] function. I have attached a notebook with detail explanation of the issue. I have run the code before and it run well. It now requires to adjust the labels of the MNIST dataset between 1 and 10, instead of the 0 to 9 in the original dataset. Is there is a way around so that no additional adjustment of the labels is needed?. In the past the adjustment was not required.

Reply to this discussion
Community posts can be styled and formatted using the Markdown syntax.
Reply Preview
Attachments
Remove
or Discard