![]() In the linear layer, you have to specify the number of input features and the number of output features which should correspond to the number of classes. The label with the highest score will be the one model predicts. In the CIFAR10 dataset, there are ten classes of labels.
0 Comments
Leave a Reply. |
AuthorWrite something about yourself. No need to be fancy, just an overview. ArchivesCategories |