Hi,everyone!
I have a problem in implementing the CNN.
“for (images,labels) in test_loader” can output the images and labels of each batch-size, but
how can save all the labels and CNN’s output of images from test_loader on whole not just one batch-size.
Thanks a lot!
If you can iterate over batches of test data, then you can save them in any way you like: python list, system files, database…?
results = []
for images, labels in test_loader: # for each batch in test data...
predictions = my_CNN(images) # get predictions for this batch
results.append([images, labels, predictions]) # keep them for further processing