smolvla-Red_Block_PicknPlace-sx1nks1pan

smolvla model - 🧪 phosphobot training pipeline

• Dataset: tremmelnicholas/Red_Block_PicknPlace
• Wandb run id: None

Error Traceback

We faced an issue while training your model.

Training process failed with exit code 1:
return forward_call(*args, **kwargs)
File "/usr/local/lib/python3.10/site-packages/transformers/models/llama/modeling_llama.py", line 172, in forward
down_proj = self.down_proj(self.act_fn(self.gate_proj(x)) * self.up_proj(x))
File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1751, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1762, in _call_impl
return forward_call(*args, **kwargs)
File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/linear.py", line 125, in forward
return F.linear(input, self.weight, self.bias)
torch.OutOfMemoryError: CUDA out of memory. Tried to allocate 88.00 MiB. GPU 0 has a total capacity of 22.06 GiB of which 51.44 MiB is free. Process 1 has 22.00 GiB memory in use. Of the allocated memory 21.10 GiB is allocated by PyTorch, and 612.88 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)

Training parameters

{
  "batch_size": 128,
  "steps": 10000,
  "save_freq": 5000
}

📖 Get Started: docs.phospho.ai

🤖 Get your robot: robots.phospho.ai