文件名称:
accelerate:训练和使用具有多GPU，TPU，混合精度的NLP模型的简单方法-源码
开发工具:
文件大小: 247kb
下载次数: 0
上传时间: 2021-03-29
详细说明：在任何类型的设备上运行* raw * PyTorch培训脚本
易于整合
:hugging_face: 为喜欢编写PyTorch模型的训练循环但不愿编写和维护使用多GPU / TPU / fp16的样板代码的PyTorch用户创建了Accelerate。
:hugging_face: 准确加速摘要，仅加速与多GPU / TPU / fp16相关的样板代码，而其余代码保持不变。
这是一个例子：
import torch
import torch.nn.functional as F
from datasets import load_dataset
+ from accelerate import Accelerator
+ accelerator = Accelerator()
- device = 'cpu'
+ device = accelerator.device
model = torch.nn.Transformer().
(系统自动生成,下载前可以参看下载内容)
下载文件列表
相关说明
- 本站资源为会员上传分享交流与学习,如有侵犯您的权益,请联系我们删除.
- 本站是交换下载平台，提供交流渠道，下载内容来自于网络，除下载问题外，其它问题请自行百度。
- 本站已设置防盗链，请勿用迅雷、QQ旋风等多线程下载软件下载资源，下载后用WinRAR最新版进行解压.
- 如果您发现内容无法下载，请稍后再次尝试；或者到消费记录里找到下载记录反馈给我们.
- 下载后发现下载的内容跟说明不相乎，请到消费记录里找到下载记录反馈给我们，经确认后退回积分.
- 如下载前有疑问，可以通过点击"提供者"的名字，查看对方的联系方式，联系对方咨询.
相关搜索:
