pepijn223 HF Staff commited on
Commit
f40f32f
·
verified ·
1 Parent(s): c45ca1b

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -14
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_base \
42
- --policy.pretrained_path=your_repo_id \
43
  --policy.compile_model=true \
44
  --policy.gradient_checkpointing=true \
45
  --wandb.enable=true \
@@ -50,18 +50,6 @@ python src/lerobot/scripts/train.py \
50
  --batch_size=32
51
  ```
52
 
53
- ## Conversion Details
54
-
55
- This model was converted from JAX to PyTorch using the OpenPI conversion script:
56
-
57
- ```bash
58
- python examples/convert_jax_model_to_pytorch.py \
59
- --checkpoint_dir /pi05_base \
60
- --config_name pi05_base \
61
- --output_path /pi05_base/pytorch/fp32/ \
62
- --precision float32
63
- ```
64
-
65
  ## Citation
66
 
67
  If you use this model, please cite the original OpenPI work:
 
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_base \
43
  --policy.compile_model=true \
44
  --policy.gradient_checkpointing=true \
45
  --wandb.enable=true \
 
50
  --batch_size=32
51
  ```
52
 
 
 
 
 
 
 
 
 
 
 
 
 
53
  ## Citation
54
 
55
  If you use this model, please cite the original OpenPI work: