文章详情

短信预约-IT技能 免费直播动态提醒

请输入下面的图形验证码

提交验证

短信预约提醒成功

优化Pytorch模型训练的小技巧

2024-12-03 08:50

关注

[[389579]]

混合精度

在一个常规的训练循环中,PyTorch以32位精度存储所有浮点数变量。对于那些在严格的约束下训练模型的人来说,这有时会导致他们的模型占用过多的内存,迫使他们使用更小的模型和更小的批处理大小进行更慢的训练过程。所以在模型中以16位精度存储所有变量/数字可以改善并修复大部分这些问题,比如显著减少模型的内存消耗,加速训练循环,同时仍然保持模型的性能/精度。

在Pytorch中将所有计算转换为16位精度非常简单,只需要几行代码。这里是:

 

  1. scaler = torch.cuda.amp.GradScaler() 

上面的方法创建一个梯度缩放标量,以最大程度避免使用fp16进行运算时的梯度下溢。

 

  1. optimizer.zero_grad() 
  2. with torch.cuda.amp.autocast(): 
  3.    output = model(input).to(device) 
  4.    loss = criterion(output, correct_answer).to(device) 
  5. scaler.scale(loss).backward() 
  6. scaler.step(optimizer) 
  7. scaler.update() 

 

当使用loss和优化器进行反向传播时,您需要使用scale .scale(loss),而不是使用loss.backward()和optimizer.step()。使用scaler.step(optimizer)来更新优化器。这允许你的标量转换所有的梯度,并在16位精度做所有的计算,最后用scaler.update()来更新缩放标量以使其适应训练的梯度。

当以16位精度做所有事情时,可能会有一些数值不稳定,导致您可能使用的一些函数不能正常工作。只有某些操作在16位精度下才能正常工作。具体可参考官方的文档。

进度条

有一个进度条来表示每个阶段的训练完成的百分比是非常有用的。为了获得进度条,我们将使用tqdm库。以下是如何下载并导入它:

 

  1. pip install tqdm 
  2. from tqdm import tqdm 

 

在你的训练和验证循环中,你必须这样做:

 

  1. for index, batch in tqdm(enumerate(loader), total = len(loader), position = 0, leave = True): 

训练和验证循环添加tqdm代码后将得到一个进度条,它表示您的模型完成的训练的百分比。它应该是这样的:

 

 

在图中,691代表我的模型需要完成多少批,7:28代表我的模型在691批上的总时间,1.54 it/s代表我的模型在每批上花费的平均时间。

梯度积累

如果您遇到CUDA内存不足的错误,这意味着您已经超出了您的计算资源。为了解决这个问题,你可以做几件事,包括把所有东西都转换成16位精度,减少模型的批处理大小,更换更小的模型等等。

但是有时切换到16位精度并不能完全解决问题。解决这个问题最直接的方法是减少批处理大小,但是假设您不想减少批处理大小可以使用梯度累积来模拟所需的批大小。请注意,CUDA内存不足问题的另一个解决方案是简单地使用多个GPU,但这是一个很多人无法使用的选项。

假设你的机器/模型只能支持16的批处理大小,增加它会导致CUDA内存不足错误,并且您希望批处理大小为32。梯度累加的工作原理是:以16个批的规模运行模型两次,将计算出的每个批的梯度累加起来,最后在这两次前向传播和梯度累加之后执行一个优化步骤。

要理解梯度积累,重要的是要理解在训练神经网络时所做的具体功能。假设你有以下训练循环:

 

  1. model = model.train() 
  2. for index, batch in enumerate(train_loader): 
  3.     input = batch[0].to(device) 
  4.     correct_answer = batch[1].to(device) 
  5.     optimizer.zero_grad() 
  6.     output = model(input).to(device) 
  7.     loss = criterion(output, correct_answer).to(device) 
  8.     loss.backward() 
  9.     optimizer.step() 

 

看看上面的代码,需要记住的关键是loss.backward()为模型创建并存储梯度,而optimizer.step()实际上更新权重。在如果在调用优化器之前两次调用loss.backward()就会对梯度进行累加。下面是如何在PyTorch中实现梯度累加:

 

  1. model = model.train() 
  2. optimizer.zero_grad() 
  3. for index, batch in enumerate(train_loader): 
  4.     input = batch[0].to(device) 
  5.     correct_answer = batch[1].to(device) 
  6.     output = model(input).to(device) 
  7.     loss = criterion(output, correct_answer).to(device) 
  8.     loss.backward() 
  9.     if (index+1) % 2 == 0: 
  10.        optimizer.step() 
  11.        optimizer.zero_grad() 

 

在上面的例子中,我们的机器只能支持16批大小的批量,我们想要32批大小的批量,我们本质上计算2批的梯度,然后更新实际权重。这导致有效批大小为32。

译者注:梯度累加只是一个折中方案,经过我们的测试,如果对梯度进行累加,那么最后一次loss.backward()的梯度会比前几次反向传播的权重高,具体为什么我们也不清楚,哈。虽然有这样的问题,但是使用这种方式进行训练还是有效果的。

16位精度的梯度累加非常类似。

 

  1. model = model.train() 
  2. optimizer.zero_grad() 
  3. for index, batch in enumerate(train_loader): 
  4.     input = batch[0].to(device) 
  5.     correct_answer = batch[1].to(device) 
  6.     with torch.cuda.amp.autocast(): 
  7.          output = model(input).to(device) 
  8.          loss = criterion(output, correct_answer).to(device) 
  9.     scaler.scale(loss).backward() 
  10.     if (index+1) % 2 == 0: 
  11.        scaler.step(optimizer) 
  12.        scaler.update() 
  13.        optimizer.zero_grad() 

 

结果评估

在大多数机器学习项目中,人们倾向于手动计算用于评估的指标。尽管计算准确率、精度、召回率和F1等指标并不困难,但在某些情况下,您可能希望拥有这些指标的某些变体,如加权精度、召回率和F1。计算这些可能需要更多的工作,如果你的实现可能不正确、高效、快速且无错误地计算所有这些指标,可以使用sklearns classification_report库。这是一个专门为计算这些指标而设计的库。

 

  1. from sklearn.metrics import classification_report 
  2. y_pred = [0, 1, 0, 0, 1] 
  3. y_correct = [1, 1, 0, 1, 1]print(classification_report(y_correct, y_pred)) 

 

上面的代码用于二进制分类。你可以为更多的目的配置这个函数。第一个列表表示模型的预测,第二个列表表示正确数值。上面的代码将输出:

 

 

结论

在这篇文章中,我讨论了4种pytorch中优化深度神经网络训练的方法。16位精度减少内存消耗,梯度积累可以通过模拟使用更大的批大小,tqdm进度条和sklearns的classification_report两个方便的库,可以轻松地跟踪模型的训练和评估模型的性能。就我个人而言,我总是用上面所有的训练技巧来训练我的神经网络,并且在必要的时候我使用梯度积累。

最后,如果你使用的是pytorch或者是pytorch的初学者,可以使用这个库:

github/deephub-ai/torch-handle

他会对你有很大的帮助。

来源:今日头条内容投诉

免责声明:

① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的,并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据,供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。

② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341

软考中级精品资料免费领

  • 历年真题答案解析
  • 备考技巧名师总结
  • 高频考点精准押题
  • 2024年上半年信息系统项目管理师第二批次真题及答案解析(完整版)

    难度     813人已做
    查看
  • 【考后总结】2024年5月26日信息系统项目管理师第2批次考情分析

    难度     354人已做
    查看
  • 【考后总结】2024年5月25日信息系统项目管理师第1批次考情分析

    难度     318人已做
    查看
  • 2024年上半年软考高项第一、二批次真题考点汇总(完整版)

    难度     435人已做
    查看
  • 2024年上半年系统架构设计师考试综合知识真题

    难度     224人已做
    查看

相关文章

发现更多好内容

猜你喜欢

AI推送时光机
位置:首页-资讯-后端开发
咦!没有更多了?去看看其它编程学习网 内容吧
首页课程
资料下载
问答资讯