pytorch -- GPU优化写法套路
时间:2024-06-18 21:53:31 点击次数:
我们关于“ 文章介绍了该软件包,并提供了背景信息。
Pytorch Forecasting旨在通过神经网络简化实际案例和研究的最新时间序列预测。目标是为高级专业人员提供最大程度的灵活性,并为初学者提供合理的默认值的高级API。具体来说,该软件包提供了
一个时间序列数据集类,它抽象化处理变量转换,缺失值,随机子采样,多个历史记录长度等。
基本模型类,提供时间序列模型的基本训练,以及在张量板中的记录和通用可视化,例如实际与预测以及依存关系图
用于时间序列预测的多种神经网络体系结构已针对实际部署进行了增强,并具有内置的解释功能
多地平线时间序列指标
Ranger优化器,用于更快的模型训练
使用调整
该程序包基于构建,可以直接使用CPU,单个和多个GPU进行培训。
安装
如果您在Windows上工作,则需要先使用以下命令安装PyTorch:
pip install torch -f https