19 lines
464 B
Bash
19 lines
464 B
Bash
|
#!/bin/bash
|
||
|
|
set -e
|
||
|
|
|
||
|
|
ROUNDS="${1:-1}"
|
||
|
|
NODES="${2:-2}"
|
||
|
|
|
||
|
|
cd "$(dirname "$0")"
|
||
|
|
|
||
|
|
mkdir -p ../reports
|
||
|
|
|
||
|
|
docker compose -f docker-compose.cpu.yml run \
|
||
|
|
-v "$(pwd)/../reports:/app/reports" \
|
||
|
|
scheduler \
|
||
|
|
bash -c "python -m sched run --script jobs/mnist.py --compose --trace --rounds $ROUNDS --nodes $NODES --report reports/report.html 2>&1 | tee reports/log.txt"
|
||
|
|
|
||
|
|
docker compose -f docker-compose.cpu.yml down
|
||
|
|
|
||
|
|
echo "Done. See reports/report.html and reports/log.txt"
|