LegrandFrederic commited on
Commit
5450571
·
verified ·
1 Parent(s): adb6d19

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +41 -0
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)