HelloCephalopod commited on
Commit
36fd5f1
·
verified ·
1 Parent(s): d77608b

Add comprehensive model card with all checkpoints

Browse files
Files changed (1) hide show
  1. README.md +61 -0
README.md ADDED
@@ -0,0 +1,61 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ base_model: HuggingFaceTB/SmolVLA
4
+ tags:
5
+ - robotics
6
+ - smolvla
7
+ - imitation-learning
8
+ - vision-language-action
9
+ - fine-tuned
10
+ library_name: lerobot
11
+ ---
12
+
13
+ # SmolVLA Fine-tuned Model
14
+
15
+ This model is a fine-tuned version of [HuggingFaceTB/SmolVLA](https://huggingface.co/HuggingFaceTB/SmolVLA) trained using the LeRobot library.
16
+
17
+ ## Model Details
18
+
19
+ - **Base Model**: HuggingFaceTB/SmolVLA
20
+ - **Training Steps**: 20,000
21
+ - **Batch Size**: 10
22
+ - **Checkpoints Available**: 4
23
+
24
+ ## Available Checkpoints
25
+
26
+ This repository contains multiple checkpoints from the training process:
27
+
28
+ - `checkpoint-020000/` - Step 020000
29
+ - `checkpoint-040000/` - Step 040000
30
+ - `checkpoint-060000/` - Step 060000
31
+ - `checkpoint-last/` - Step last
32
+
33
+ ## Usage
34
+
35
+ ```python
36
+ from lerobot import LeRobotPolicy
37
+
38
+ # Load the final checkpoint
39
+ model = LeRobotPolicy.from_pretrained("HelloCephalopod/my_smolvla_2/checkpoint-last")
40
+
41
+ # Or load a specific checkpoint
42
+ model = LeRobotPolicy.from_pretrained("HelloCephalopod/my_smolvla_2/checkpoint-10000")
43
+ ```
44
+
45
+ ## Training Details
46
+
47
+ - **Training Data**: Custom dataset from Hugging Face Hub
48
+ - **Training Steps**: 60,000
49
+ - **Batch Size**: 10
50
+ - **Save Steps**: 20,000
51
+ - **Hardware**: Google Colab GPU
52
+
53
+ ## Checkpoint Selection
54
+
55
+ - **checkpoint-40000**: Mid-training checkpoint
56
+ - **checkpoint-60000**: Final checkpoint
57
+ - **checkpoint-last**: Latest checkpoint (recommended)
58
+
59
+ ## Citation
60
+
61
+ If you use this model, please cite the original SmolVLA paper and the LeRobot library.