算力平台:
从这里开始!
请使用下面的交互式工具来帮助你开始了解 Accelerate 的某个特定功能及其使用方法!它将为你提供代码差异、解释以及一些有用的链接,以便你在文档中进一步探索!
大多数代码示例在集成 Accelerate 之前都从以下 Python 代码开始:
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()