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

发布于 2022-09-12 02:28:49 字数 321 浏览 33 评论 0

目前只会先把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)
        

如果你对这篇内容有疑问,欢迎到本站社区发帖提问 参与讨论,获取更多帮助,或者扫码二维码加入 Web 技术交流群。

扫码二维码加入Web技术交流群

发布评论

需要 登录 才能够评论, 你可以免费 注册 一个本站的账号。

评论(1

紙鸢 2022-09-19 02:28:49

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

~没有更多了~
我们使用 Cookies 和其他技术来定制您的体验包括您的登录状态等。通过阅读我们的 隐私政策 了解更多相关信息。 单击 接受 或继续使用网站,即表示您同意使用 Cookies 和您的相关数据。
原文