摘要: 梯度裁剪(Gradient Clipping) import torch.nn as nn outputs = model(data) loss= loss_fn(outputs, target) optimizer.zero_grad() loss.backward() nn.utils.clip 阅读全文
posted @ 2019-11-25 22:10 爱学英语的程序媛 阅读(1497) 评论(0) 推荐(0) 编辑