Commencez ici !

Veuillez utiliser l’outil interactif ci-dessous pour vous aider à démarrer l’apprentissage d’une fonctionnalité particulière d’Accelerate et de sa mise en œuvre ! Il vous fournira un diff de code, une explication de ce qui se passe, ainsi que des liens utiles pour explorer davantage la documentation !

La plupart des exemples de code partent du code Python suivant avant d’intégrer Accelerate d’une manière ou d’une autre :

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()