... batch_idx): x, y = batch y_hat = self(x) labels_hat = torch.argmax(y_hat, dim=1) n_correct_pred = torch.sum(y == labels_hat).item() loss ...
確定! 回上一頁