You are viewing v0.31.0 version.
A newer version
v1.2.1 is available.
Learning how to incorporate π€ Accelerate features quickly!
Please use the interactive tool below to help you get started with learning about a particular feature of π€ Accelerate and how to utilize it! It will provide you with a code diff, an explanation towards what is going on, as well as provide you with some useful links to explore more within the documentation!
Most code examples start from the following python code before integrating π€ Accelerate in some way:
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()