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