Accelerate 文档
开始这里!
加入 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()