you would have to use a Dataset definition that more closely resembles one designed for multi-class classification than for binary classification. The example code presented in this article can be ...

This article is the third in a series of four articles that present a complete end-to-end production-quality example of binary classification using a PyTorch ... The demo begins by creating Dataset ...