登录 注册
当前位置:主页 > 资源下载 > 32 > 在PyTorch上构建贝叶斯神经网络层,blitz-bayesian-deep-learning是一个易于使用且具备扩展性的库

在PyTorch上构建贝叶斯神经网络层,blitz-bayesian-deep-learning是一个易于使用且具备扩展性的库

  • 更新:2024-05-19 15:26:20
  • 大小:136KB
  • 推荐:★★★★★
  • 来源:网友上传分享
  • 类别:其它 - 开发技术
  • 格式:ZIP

资源介绍

闪电战-火炬动物园中的贝叶斯层 BLiTZ是一个简单且可扩展的库,用于在PyTorch上创建贝叶斯神经网络层(基于“)。 通过使用BLiTZ图层和utils,您可以以不影响图层之间的交互的简单方式(例如,就像使用标准PyTorch一样)添加非证书并收集模型的复杂性成本。 通过使用我们的核心权重采样器类,您可以扩展和改进此库,从而以与PyTorch良好集成的方式为更大范围的图层添加不确定性。 也欢迎拉取请求。 我们的目标是使人们能够通过专注于他们的想法而不是硬编码部分来应用贝叶斯深度学习。 Rodamap: 为不同于正态的后验分布启用重新参数化。 指数 贝叶斯层的目的 贝叶斯层上的权重采样 有可能优化我们的可训练重量 的确,存在复杂度成本函数随其变量可微分的情况。 在第n个样本处获得整个成本函数 一些笔记和总结 引用 参考 安装 要安装BLiTZ,可以使用pip命令: pip