Pytorch模型test时候显存释放