Upload README.md with huggingface_hub
Browse files
README.md
ADDED
@@ -0,0 +1,41 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
|
2 |
+
---
|
3 |
+
tags:
|
4 |
+
- phosphobot
|
5 |
+
- act
|
6 |
+
task_categories:
|
7 |
+
- robotics
|
8 |
+
---
|
9 |
+
|
10 |
+
# act Model - phospho Training Pipeline
|
11 |
+
|
12 |
+
|
13 |
+
## Error Traceback
|
14 |
+
We faced an issue while training your model.
|
15 |
+
|
16 |
+
```
|
17 |
+
Training process failed with exit code 1:
|
18 |
+
return forward_call(*args, **kwargs)
|
19 |
+
File "/lerobot/lerobot/common/policies/act/modeling_act.py", line 583, in forward
|
20 |
+
x = self.linear2(self.dropout(self.activation(self.linear1(x))))
|
21 |
+
File "/opt/conda/envs/lerobot/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1739, in _wrapped_call_impl
|
22 |
+
return self._call_impl(*args, **kwargs)
|
23 |
+
File "/opt/conda/envs/lerobot/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1750, in _call_impl
|
24 |
+
return forward_call(*args, **kwargs)
|
25 |
+
File "/opt/conda/envs/lerobot/lib/python3.10/site-packages/torch/nn/modules/linear.py", line 125, in forward
|
26 |
+
return F.linear(input, self.weight, self.bias)
|
27 |
+
torch.OutOfMemoryError: CUDA out of memory. Tried to allocate 736.00 MiB. GPU 0 has a total capacity of 22.07 GiB of which 626.25 MiB is free. Process 17 has 21.45 GiB memory in use. Of the allocated memory 20.55 GiB is allocated by PyTorch, and 640.91 MiB is reserved by PyTorch but unallocated. If reserved but unallocated memory is large try setting PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True to avoid fragmentation. See documentation for Memory Management (https://pytorch.org/docs/stable/notes/cuda.html#environment-variables)
|
28 |
+
```
|
29 |
+
|
30 |
+
|
31 |
+
## Training parameters:
|
32 |
+
|
33 |
+
- **Dataset**: [anthonyav/so100-lego-v2](https://huggingface.co/datasets/anthonyav/so100-lego-v2)
|
34 |
+
- **Wandb run URL**: None
|
35 |
+
- **Epochs**: None
|
36 |
+
- **Batch size**: 100
|
37 |
+
- **Training steps**: 8000
|
38 |
+
|
39 |
+
📖 **Get Started**: [docs.phospho.ai](https://docs.phospho.ai?utm_source=replicate_groot_training_pipeline)
|
40 |
+
|
41 |
+
🤖 **Get your robot**: [robots.phospho.ai](https://robots.phospho.ai?utm_source=replicate_groot_training_pipeline)
|