File size: 286 Bytes
f6d8768 | 1 2 3 4 5 6 7 8 9 10 11 | #!/usr/bin/env bash
set -euo pipefail
output_dir="${MINIDREAMER_OUTPUT_DIR:-artifacts/world_model_experiment}"
export PYTHONPATH="${PYTHONPATH:-}:$(pwd)/src"
python3.11 src/train_world_model.py \
--config configs/fourrooms_world_model.yaml \
--output-dir "${output_dir}" \
"$@"
|