训练没问题测试报内存溢出：RuntimeError: CUDA out of memory...

OOM报错如图所示：

训练跑得好好的，怎么测试的时候反倒报错了呢？
原来是没有加with torch.no_grad()，在测试的时候有gradient容易报内存溢出，加上：

with torch.no_grad():
	testY = model(testX)

再跑代码就OK了。