Accelerate 文档

开始这里!

Hugging Face's logo
加入 Hugging Face 社区

并获得增强的文档体验

开始使用

开始这里!

请使用下面的交互式工具,以帮助你开始学习 Accelerate 的特定功能以及如何使用它!它将为你提供代码差异、对正在发生的事情的解释,以及一些有用的链接,以便在文档中进行更多探索!

大多数代码示例从以下 python 代码开始,然后再以某种方式集成 Accelerate

for batch in dataloader:
    optimizer.zero_grad()
    inputs, targets = batch
    inputs = inputs.to(device)
    targets = targets.to(device)
    outputs = model(inputs)
    loss = loss_function(outputs, targets)
    loss.backward()
    optimizer.step()
    scheduler.step()
< > 在 GitHub 上更新