pytorch有没有一下子送进去data_set做test accuracy的简洁写法？

• Z时代
• 2024-02-21
• 分类：IT

目前只会先把test_set放到data_loader里面，让batch-size等于len(test_set)。这样一次就可全放入。
但是也没有更简洁的写法？

    data_loader=torch.utils.data.DataLoader(test_set,batch_size=len(test_set))
    for x,labels in data_loader:
        preds=global_net(x)
        num=GetCorrectNum(preds,labels)

回答：

2021年7月。看了当时的我的问题，按照我目前的知识储备来看，确实是最简单的了。
torch并没有提供一个单独的test dataset，能够直接把所有的数据一股脑送进去model。能且只能用batch的方法单批送进去。

以上是 pytorch有没有一下子送进去data_set做test accuracy的简洁写法？ 的全部内容， 来源链接： utcz.com/p/937822.html