docs: update documentation for logging and configuration
- Add Quick Start guide with installation and configuration steps - Document unified logging framework features - Explain configuration management for multiple researchers - Add project structure overview showing experiment_logger - Update training examples with new patterns - Add logging best practices to CONTRIBUTING.md
This commit is contained in:
parent
f31436bccd
commit
7fe217de0f
2 changed files with 78 additions and 5 deletions
74
README.md
74
README.md
|
|
@ -1,17 +1,81 @@
|
|||
# Brittle Star
|
||||
|
||||
## Usage
|
||||
Reinforcement learning research on brittle star locomotion using PPO.
|
||||
|
||||
### UV
|
||||
## Quick Start
|
||||
|
||||
To set up the UV module, you can run the following command:
|
||||
### Installation
|
||||
|
||||
Set up the environment using UV:
|
||||
|
||||
```bash
|
||||
uv sync --frozen
|
||||
```
|
||||
|
||||
example command:
|
||||
### Configuration
|
||||
|
||||
1. **Copy the default configuration:**
|
||||
```bash
|
||||
cp configs/default_ppo.yaml configs/my_experiment.yaml
|
||||
```
|
||||
|
||||
2. **Edit `configs/my_experiment.yaml`** to set your WandB credentials:
|
||||
```yaml
|
||||
track: true # Enable WandB logging
|
||||
wandb_entity: "your-wandb-username" # Replace with your username/team
|
||||
wandb_project_name: "PPO-Modularity"
|
||||
```
|
||||
|
||||
3. **(Optional) Login to WandB:**
|
||||
```bash
|
||||
uv run wandb login
|
||||
```
|
||||
|
||||
### Training
|
||||
|
||||
Run training with your configuration:
|
||||
|
||||
```bash
|
||||
uv run src/train.py --model_name my_model --epochs 50 --batch_size 32
|
||||
uv run python src/train.py
|
||||
```
|
||||
|
||||
Or use a custom config file:
|
||||
|
||||
```bash
|
||||
uv run python src/train.py --config configs/my_experiment.yaml
|
||||
```
|
||||
|
||||
Override specific parameters:
|
||||
|
||||
```bash
|
||||
uv run python src/train.py --learning-rate 0.001 --num-envs 32 --track
|
||||
```
|
||||
|
||||
### Logging
|
||||
|
||||
The training script uses a unified logging framework that:
|
||||
- Logs to **WandB** (when enabled)
|
||||
- Saves metrics to **local disk** (JSON files in `runs/`)
|
||||
- Displays progress in **stdout**
|
||||
|
||||
All experiment data is preserved locally, even if WandB is unavailable.
|
||||
|
||||
## Project Structure
|
||||
|
||||
```
|
||||
src/brittle_star_project/ # Core library (reusable components)
|
||||
├── logging/ # Unified logging framework
|
||||
├── environment/ # Environment wrappers
|
||||
├── rl/ # RL algorithms and models
|
||||
└── dataclasses/ # Configuration dataclasses
|
||||
|
||||
configs/ # Training configurations
|
||||
runs/ # Training outputs (checkpoints, metrics)
|
||||
```
|
||||
|
||||
## For Researchers
|
||||
|
||||
**Important:** Do not commit your personal WandB credentials to the repository.
|
||||
Instead, create your own config file (e.g., `configs/yourname.yaml`) and add it to `.gitignore` if needed.
|
||||
|
||||
See [configs/README.md](configs/README.md) for more details on configuration management.
|
||||
|
|
|
|||
Reference in a new issue