pytorch的辅助功能模块

1. torch.utils.bottleneck模块

torch.utils.bottleneck可以用来检查深度学习模型中的运行时间,从而可以找到导致性能瓶颈的那些模块,并优化那些模块的运行时间,进而优化整个深度学习模型的性能。
例:

python -m torch.utils.bottleneck /path/to/source/script.py [args]

2. torch.utils.checkpoint模块

torch.checkpoint可以用来节约深度学习使用的内存。因为要进行梯度反传播,所以在构建计算图的时候需要保存中间的数据,而这数据大大增加了深度学习的内存消耗。我们可以通过该模块记录中间数据的计算过程,然后丢弃这些中间数据,等需要使用时再重新计算这些数据,这样就减少了内存消耗,进而可以提升深度学习模型的性能和优化时的稳定性。该模块设计的核心

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

饕餮&化骨龙

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值