You are viewing v0.25.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()