pepijn223 HF Staff commited on
Commit
d3e8ab4
·
verified ·
1 Parent(s): 82cba76

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -13
README.md CHANGED
@@ -38,8 +38,8 @@ python src/lerobot/scripts/train.py \
38
  --policy.type=pi05 \
39
  --output_dir=./outputs/pi05_training \
40
  --job_name=pi05_training \
41
- --policy.repo_id=pepijn223/pi05_libero \
42
- --policy.pretrained_path=your_repo_id \
43
  --policy.compile_model=true \
44
  --policy.gradient_checkpointing=true \
45
  --wandb.enable=true \
@@ -49,17 +49,6 @@ python src/lerobot/scripts/train.py \
49
  --policy.device=cuda \
50
  --batch_size=32
51
  ```
52
- ## Conversion Details
53
-
54
- This model was converted from JAX to PyTorch using the OpenPI conversion script:
55
-
56
- ```bash
57
- python examples/convert_jax_model_to_pytorch.py \
58
- --checkpoint_dir /pi05_libero \
59
- --config_name pi05_libero \
60
- --output_path /pi05_base/pytorch/fp32/ \
61
- --precision float32
62
- ```
63
 
64
  ## Citation
65
 
 
38
  --policy.type=pi05 \
39
  --output_dir=./outputs/pi05_training \
40
  --job_name=pi05_training \
41
+ --policy.repo_id=your_repo_id \
42
+ --policy.pretrained_path=lerobot/pi05_libero_base \
43
  --policy.compile_model=true \
44
  --policy.gradient_checkpointing=true \
45
  --wandb.enable=true \
 
49
  --policy.device=cuda \
50
  --batch_size=32
51
  ```
 
 
 
 
 
 
 
 
 
 
 
52
 
53
  ## Citation
54