@torch.no_grad()

with torch.no_grad()或者@torch.no_grad()中的数据不需要计算梯度,也不会进行反向传播

model.eval()                               
with torch.no_grad():
   ...

等价于

@torch.no_grad()
def eval():
	...

你可能感兴趣的:(python,人工智能,机器学习,深度学习)