Environment setup + train loop (#5)
Mujoco environment setup (vectorized on GPU) + training loop + simulate script
This commit is contained in:
parent
af9cf1fdc5
commit
a85a7b8d89
27 changed files with 2155 additions and 234 deletions
17
README.md
17
README.md
|
|
@ -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
|
||||
```
|
||||
Reference in a new issue