pytorch有没有一下子送进去data_set做test accuracy的简洁写法?
目前只会先把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 技术交流群。
绑定邮箱获取回复消息
由于您还没有绑定你的真实邮箱,如果其他用户或者作者回复了您的评论,将不能在第一时间通知您!
发布评论
评论(1)
2021年7月。看了当时的我的问题,按照我目前的知识储备来看,确实是最简单的了。
torch并没有提供一个单独的test dataset,能够直接把所有的数据一股脑送进去model。能且只能用batch的方法单批送进去。