batch_size=32 train_data_loader = torch.utils.data. ... target= target.to(device) output = model(input) correct = torch.eq(torch.max(output, dim=1)[1], ...
確定! 回上一頁