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)
        
阅读 1.3k
1 个回答

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

撰写回答
你尚未登录,登录后可以
  • 和开发者交流问题的细节
  • 关注并接收问题和回答的更新提醒
  • 参与内容的编辑和改进,让解决方法与时俱进
推荐问题
宣传栏