pytorch在推理时出现CUDA error: out of memory

pytorch在推理时出现CUDA error: out of memory

发现是这个问题

with torch.no_grad():
	output = net(input)

你可能感兴趣的:(报错bug,python,bug,pytorch,神经网络,深度学习)