1
Fork 0

Environment setup + train loop (#5)

Mujoco environment setup (vectorized on GPU) + training loop + simulate script
This commit is contained in:
RobinMeersman 2026-03-26 09:53:54 +01:00 committed by GitHub
parent af9cf1fdc5
commit a85a7b8d89
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
27 changed files with 2155 additions and 234 deletions

View file

@ -0,0 +1,17 @@
# Brittle Star
## Usage
### UV
To set up the UV module, you can run the following command:
```bash
uv sync --frozen
```
example command:
```bash
uv run src/train.py --model_name my_model --epochs 50 --batch_size 32
```