原因在于没有使用torch.no_grad()函数。在查看验证集和测试集表现时,应使用类似这样的代码
def evaluate(data_loader):
with torch.no_grad():
mean_acc, mean_iou = 0, 0
for i, (img, gnd) in enumerate(data_loader):
if torch.cuda.is_available():
img = img.cuda(device=device)
gnd = gnd.cuda(device=device)
out = model(img)
.......
return mean_acc / len(data_loader), mean_iou / len(data_loader)
原文地址:https://www.cnblogs.com/liuzhan709/p/10053009.html
时间: 2024-10-27 11:18:41