File size: 692 Bytes
09dd617 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 | #!/bin/bash
# Run DPA simulation on local machine (CPU/MPS, no GPU needed)
set -e
cd "$(dirname "$0")/.."
echo "=== Generating agent trajectory dataset ==="
python3 src/data/agent_trajectory.py
echo ""
echo "=== Running DPA simulation benchmark ==="
python3 src/eval/benchmark.py \
--mode simulate \
--hidden-size 512 \
--num-layers 6 \
--num-heads 8 \
--seq-len 1024 \
--batch-size 4 \
--output-dir results/simulation
echo ""
echo "=== Generating figures ==="
python3 src/eval/visualize.py results/simulation/simulation_results.json
python3 src/eval/visualize.py data/agent_trajectories.json # trajectory analysis
echo ""
echo "=== Done! Check results/ and figures/ ==="
|