Update README.md
Browse files
README.md
CHANGED
|
@@ -75,7 +75,7 @@ For reproducibility, here are the key parameters used during this initial traini
|
|
| 75 |
- **Scheduler**: Cosine with warmup
|
| 76 |
- **Vision**: ResNet18 with random crop (84x84)
|
| 77 |
|
| 78 |
-
#### Original Training Command
|
| 79 |
|
| 80 |
```bash
|
| 81 |
python -m lerobot.scripts.lerobot_train \
|
|
@@ -92,6 +92,18 @@ python -m lerobot.scripts.lerobot_train \
|
|
| 92 |
|
| 93 |
## 🚀 Evaluate (My Evaluation Mode)
|
| 94 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 95 |
You can evaluate this checkpoint to reproduce the Phase 1 results:
|
| 96 |
|
| 97 |
```bash
|
|
|
|
| 75 |
- **Scheduler**: Cosine with warmup
|
| 76 |
- **Vision**: ResNet18 with random crop (84x84)
|
| 77 |
|
| 78 |
+
#### Original Training Command (My Training Mode)
|
| 79 |
|
| 80 |
```bash
|
| 81 |
python -m lerobot.scripts.lerobot_train \
|
|
|
|
| 92 |
|
| 93 |
## 🚀 Evaluate (My Evaluation Mode)
|
| 94 |
|
| 95 |
+
Run the following command in your terminal to evaluate the model for 50 episodes and save the visualization videos:
|
| 96 |
+
|
| 97 |
+
```bash
|
| 98 |
+
python -m lerobot.scripts.lerobot_eval \
|
| 99 |
+
--policy.type diffusion \
|
| 100 |
+
--policy.pretrained_path outputs/train/2025-12-02/14-33-35_DP_PushT/checkpoints/last/pretrained_model \
|
| 101 |
+
--eval.n_episodes 50 \
|
| 102 |
+
--eval.batch_size 10 \
|
| 103 |
+
--env.type pusht \
|
| 104 |
+
--env.task PushT-v0
|
| 105 |
+
```
|
| 106 |
+
|
| 107 |
You can evaluate this checkpoint to reproduce the Phase 1 results:
|
| 108 |
|
| 109 |
```bash
|