requires_grad=True
: 要求计算梯度requires_grad=False
: 不要求计算梯度with torch.no_grad()
或者@torch.no_grad()
中的数据不需要计算梯度,也不会进行反向传播。
model.eval() # 测试模式
with torch.no_grad():
pass
@torch.no_grad()
def eval():
...
requires_grad=True
: 要求计算梯度requires_grad=False
: 不要求计算梯度with torch.no_grad()
或者@torch.no_grad()
中的数据不需要计算梯度,也不会进行反向传播。model.eval() # 测试模式
with torch.no_grad():
pass
@torch.no_grad()
def eval():
...