Configuration
YAML Config File
WSDP supports configuration via YAML files:
# config.yaml
widar:
learning_rate: 0.001
num_epochs: 50
batch_size: 64
gait:
learning_rate: 0.0005
num_epochs: 100
Usage:
CLI Parameters
All hyperparameters can be overridden via CLI:
| Parameter | CLI Flag | Default |
|---|---|---|
| Learning Rate | --lr |
From model_params.json |
| Epochs | --epochs |
From model_params.json |
| Batch Size | --batch-size |
From model_params.json |
| Config File | --config |
None |