train.py的test部分是不是有BUG，使用的时候会爆显存

[Big-Orange-Tiger]

    for idx_iter, (img, gt_mask, size, _) in enumerate(test_loader):
        img = Variable(img).cuda()
        pred = net.forward(img)
        pred = pred[:,:,:size[0],:size[1]]
        gt_mask = gt_mask[:,:,:size[0],:size[1]]
        eval_mIoU.update((pred>opt.threshold).cpu(), gt_mask)
        eval_PD_FA.update((pred[0,0,:,:]>opt.threshold).cpu(), gt_mask[0,0,:,:], size)

这段循环之前没有类似test.py之中的with torch.no_grad():是不是此处出现了BUG

[XinyiYing]

需要加上with torch.no_grad():，同时参考#issues32