Zouzouw commited on
Commit
d253741
·
verified ·
1 Parent(s): 0ef9856

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +25 -12
README.md CHANGED
@@ -12,28 +12,42 @@ pipeline_tag: robotics
12
 
13
  Politique **ACT** pour le bras SO-101, entrainee par apprentissage par imitation.
14
 
15
- Ce depot est un **fine-tuning** de la base [`Zouzouw/act_fusion`](https://huggingface.co/Zouzouw/act_fusion),
16
- affinee sur [`Zouzouw/dataset_3cube3couleur`](https://huggingface.co/datasets/Zouzouw/dataset_3cube3couleur).
17
 
18
  | | |
19
  |---|---|
20
- | Base | `Zouzouw/act_fusion` (entrainee sur `Zouzouw/fusion`) |
21
  | Dataset de fine-tuning | `Zouzouw/dataset_3cube3couleur` (96 episodes) |
22
  | **Learning rate** | **1e-05** |
23
- | Checkpoint | etape 150 000 |
24
  | Politique | ACT |
25
 
 
 
26
  ## La tache
27
 
28
  Attraper trois cubes — vert, puis rouge, puis noir — et poser chacun sur la croix
29
  de sa couleur, dans cet ordre.
30
 
31
- ## Pourquoi deux fine-tunes
 
 
 
 
 
 
 
 
 
 
 
 
 
 
32
 
33
- `act_fusion_ft1` et `act_fusion_ft2` sont le meme entrainement a deux **learning rates**
34
- differents (1e-06 pour l'autre), pour chercher le bon equilibre : trop eleve, le modele
35
- oublie ce qu'il savait deja ; trop faible, il ne se specialise pas. Les comparer
36
- **sur le robot**, pas seulement sur les courbes.
37
 
38
  ## Utilisation
39
 
@@ -41,6 +55,5 @@ oublie ce qu'il savait deja ; trop faible, il ne se specialise pas. Les comparer
41
  lerobot-rollout --policy.path=Zouzouw/act_fusion_ft1 --robot.type=so101_follower ...
42
  ```
43
 
44
- Contenu : les poids seuls (`pretrained_model/`), suffisants pour l'inference.
45
- Pour reprendre l'entrainement, partir de la base `Zouzouw/act_fusion`, qui inclut
46
- l'etat de l'optimiseur.
 
12
 
13
  Politique **ACT** pour le bras SO-101, entrainee par apprentissage par imitation.
14
 
15
+ Fine-tuning de [`Zouzouw/act_fusion`](https://huggingface.co/Zouzouw/act_fusion)
16
+ sur [`Zouzouw/dataset_3cube3couleur`](https://huggingface.co/datasets/Zouzouw/dataset_3cube3couleur).
17
 
18
  | | |
19
  |---|---|
20
+ | Base | `Zouzouw/act_fusion` — ACT entrainee sur `Zouzouw/fusion` (319 episodes) |
21
  | Dataset de fine-tuning | `Zouzouw/dataset_3cube3couleur` (96 episodes) |
22
  | **Learning rate** | **1e-05** |
23
+ | Checkpoints | **200 000** (racine) et 150 000 (sous-dossier `150000/`) |
24
  | Politique | ACT |
25
 
26
+ > **C'est le modele le plus performant du projet a ce jour** — le checkpoint a l'etape 200 000, celui servi a la racine de ce depot.
27
+
28
  ## La tache
29
 
30
  Attraper trois cubes — vert, puis rouge, puis noir — et poser chacun sur la croix
31
  de sa couleur, dans cet ordre.
32
 
33
+ ## Le learning rate, et pourquoi il y a deux depots
34
+
35
+ `act_fusion_ft1` et `act_fusion_ft2` sont **le meme fine-tuning a deux learning rates**,
36
+ pour trouver le bon equilibre :
37
+
38
+ | Depot | Learning rate | |
39
+ |---|---|---|
40
+ | [`act_fusion_ft1`](https://huggingface.co/Zouzouw/act_fusion_ft1) | **1e-05** | le plus performant a ce jour |
41
+ | [`act_fusion_ft2`](https://huggingface.co/Zouzouw/act_fusion_ft2) | **1e-06** | plus prudent, se specialise moins |
42
+
43
+ Le *learning rate* regle **la vitesse a laquelle le modele revise ce qu'il sait**.
44
+ Trop eleve, il oublie la base et ne garde que la nouvelle tache. Trop faible, il ne
45
+ se specialise pas et reste sur son comportement general. La recherche du bon reglage
46
+ n'est pas terminee : d'autres valeurs restent a essayer, en repartant de la base,
47
+ qui inclut l'etat de l'optimiseur.
48
 
49
+ **Comparez sur le robot, pas seulement sur les courbes d'entrainement.** Une courbe
50
+ de perte qui descend bien ne garantit pas un meilleur geste.
 
 
51
 
52
  ## Utilisation
53
 
 
55
  lerobot-rollout --policy.path=Zouzouw/act_fusion_ft1 --robot.type=so101_follower ...
56
  ```
57
 
58
+ Contenu : les poids seuls, suffisants pour l'inference. Pour relancer un entrainement,
59
+ partir de [`Zouzouw/act_fusion`](https://huggingface.co/Zouzouw/act_fusion).