Get predictions from the maximum value predicted = torch.max(outputs.data, 1)[1] # 8.Total number of labels total_train += len(labels) # 9.
確定! 回上一頁