模型输出的维度需要与类别数相等,比如
classifier = Classifier(INPUT_DIM, HIDDEN_DIM, OUTPUT_DIM).to(DEVICE)
logit = classifier(data)
data如果有5个类,那么OUTPUT_DIM也应该设置为5
模型输出的维度需要与类别数相等,比如
classifier = Classifier(INPUT_DIM, HIDDEN_DIM, OUTPUT_DIM).to(DEVICE)
logit = classifier(data)
data如果有5个类,那么OUTPUT_DIM也应该设置为5