Closed laowangzi closed 2 years ago
Hi, Excellent work. However, I have a question. In the train section of train.py, your code isy = torch.zeros(len(y_pred)).long().to(device). The true label I think is 1 + 0*K, rather than 0 * (K + 1).
y = torch.zeros(len(y_pred)).long().to(device)
1 + 0*K
0 * (K + 1)
Hi, Excellent work. However, I have a question. In the train section of train.py, your code is
y = torch.zeros(len(y_pred)).long().to(device). The true label I think is1 + 0*K, rather than0 * (K + 1).