pytorch测试模型时显存不够的问题

在训练过程中使用了分块训练，但是在测试的时候使用了整张图片输入到网络中，然后就出现了显存不够的问题
经验告诉我不应该不够，但是使用了分块训练，我也不知道是不是真的不够
一筹莫展之时，发现没有使用torch.no_grad()，用了之后就没有出现显存不够的问题
虽然定义Tensor时默认是no_grad的，但是在使用神经网络时，参数或权重是默认需要梯度的，与之连接的参数因此也是默认需要梯度的。