从这里开始!
请使用下面的交互式工具来帮助您开始学习 Accelerate 的特定功能以及如何使用它!它将为您提供代码差异、对正在发生的事情的解释,以及提供一些有用的链接,以便您可以在文档中探索更多内容!
大多数代码示例在某种程度上集成 Accelerate 之前,都从以下 Python 代码开始
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()