In that case, just pass the class index targets into the loss function and PyTorch will take care of the rest. For the encoder, decoder and ...
確定! 回上一頁