Multiclass Classification Example In Pytorch