Closed ClonedOne closed 3 years ago
Hello and thank you for sharing your code.
I'd like to ask what is the format of the labels to use? are they assumed to be one-hot encoded or just integers?
Thank you!
The code assumes integer labels from 0 to K-1, same as when training a model in PyTorch.
Let me know if you have any other question.
Thank you very much for the clarification!
Hello and thank you for sharing your code.
I'd like to ask what is the format of the labels to use? are they assumed to be one-hot encoded or just integers?
Thank you!