Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
<pre> | |
+from accelerate import Accelerator | |
+accelerator = Accelerator() | |
+dataloader, model, optimizer scheduler = accelerator.prepare( | |
+ dataloader, model, optimizer, scheduler | |
+) | |
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() | |
+ accelerator.backward(loss) | |
optimizer.step() | |
scheduler.step()</pre> |