Accelerate documentation

Mulai di Sini!

You are viewing main version, which requires installation from source. If you'd like regular pip install, checkout the latest stable version (v1.15.0).
Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

Mulai di Sini!

Silakan gunakan alat interaktif di bawah ini untuk membantu Anda memulai belajar tentang fitur tertentu dari Accelerate dan cara memanfaatkannya! Alat ini akan memberikan Anda diff kode, penjelasan tentang apa yang sedang terjadi, serta beberapa tautan berguna untuk menjelajahi lebih dalam di dalam dokumentasi!

Sebagian besar contoh kode dimulai dari kode python berikut sebelum mengintegrasikan Accelerate dengan cara tertentu:

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()
Update on GitHub