目前只会先把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的方法单批送进去。