-
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(): ...
with torch.no_grad()或@torch.no_grad() 用法
最新推荐文章于 2024-07-19 10:46:01 发布