mirror of
https://github.com/huggingface/lerobot.git
synced 2026-05-22 12:09:42 +00:00
fix pre commit
This commit is contained in:
@@ -57,6 +57,7 @@ accelerate launch \
|
|||||||
```
|
```
|
||||||
|
|
||||||
**Key accelerate parameters:**
|
**Key accelerate parameters:**
|
||||||
|
|
||||||
- `--multi_gpu`: Enable multi-GPU training
|
- `--multi_gpu`: Enable multi-GPU training
|
||||||
- `--num_processes=2`: Number of GPUs to use
|
- `--num_processes=2`: Number of GPUs to use
|
||||||
- `--mixed_precision=fp16`: Use fp16 mixed precision (or `bf16` if supported)
|
- `--mixed_precision=fp16`: Use fp16 mixed precision (or `bf16` if supported)
|
||||||
@@ -97,4 +98,3 @@ For faster training, you can enable mixed precision (fp16 or bf16). This is conf
|
|||||||
- When saving or pushing models, LeRobot automatically unwraps the model from accelerate's distributed wrapper to ensure compatibility.
|
- When saving or pushing models, LeRobot automatically unwraps the model from accelerate's distributed wrapper to ensure compatibility.
|
||||||
|
|
||||||
For more advanced configurations and troubleshooting, see the [Accelerate documentation](https://huggingface.co/docs/accelerate).
|
For more advanced configurations and troubleshooting, see the [Accelerate documentation](https://huggingface.co/docs/accelerate).
|
||||||
|
|
||||||
|
|||||||
@@ -15,10 +15,10 @@
|
|||||||
# limitations under the License.
|
# limitations under the License.
|
||||||
import logging
|
import logging
|
||||||
import time
|
import time
|
||||||
|
from collections.abc import Callable
|
||||||
from contextlib import nullcontext
|
from contextlib import nullcontext
|
||||||
from pprint import pformat
|
from pprint import pformat
|
||||||
from typing import Any
|
from typing import Any
|
||||||
from collections.abc import Callable
|
|
||||||
|
|
||||||
import torch
|
import torch
|
||||||
from termcolor import colored
|
from termcolor import colored
|
||||||
|
|||||||
Reference in New Issue
Block a user