© 1999-2048 dssz.net 粤ICP备11031372号
[其它] pytorch梯度剪裁方式
说明:我就废话不多说,看例子吧! import torch.nn as nn outputs = model(data) loss= loss_fn(outputs, target) optimizer.zero_grad() loss.backward() nn.utils.clip_grad_norm_(model.parameters(), max_norm=20, norm_type=2) optimizer.step() nn.utils.clip_grad_norm_ 的参数: par<weixin_38700790> 上传 | 大小:33kb