在PyTorch中,为确保模型对单张图像和批量图像的预测一致,需将模型设置为评估模式并禁用梯度计算。使用model.eval()和torch.no_grad()可以减少不必要的计算,获得稳定的预测结果。
完成下面两步后,将自动完成登录并继续当前操作。