When I follow the tutorial “https://pytorch.org/tutorials/intermediate/torchvision_tutorial.html 3” and I make the number of predictions it is only 100? is that due to the setting in pytorch/torchvision, it is not the model?
img, _ =dataset_test[1]# dataset[0]
put the model in evaluation mode
model.eval()
with torch.no_grad():
prediction = model([img.to(device)])
prediction[0][“scores”]
#this only gives a 100, which is not enough for the application I will use it for.