accelerate

Uma biblioteca que habilita o mesmo código PyTorch a rodar em qualquer configuração distribuída. Mais informações: https://huggingface.co/docs/accelerate/index.

• Mostra informações do ambiente:

accelerate env

• Cria um arquivo de configuração de forma interativa:

accelerate config

• Mostra o custo de memória de GPU estimado para rodar um Modelo de Face aumentado com diferentes tipos de dados:

accelerate estimate-memory {{nome/modelo}}

• Testa um arquivo Accelerate de configuração:

accelerate test --config_file {{caminho/para/config.yaml}}

• Roda um modelo na CPU com Accelerate:

accelerate launch {{caminho/para/script.py}} {{--cpu}}

• Roda um modelo em multi-GPU com Accelerate, com 2 máquinas:

accelerate launch {{caminho/para/script.py}} --multi_gpu --num_machines 2