test_ci.sh

#!/bin/bash
set -euxo pipefail

pip install -r requirements.txt
colossalai run --nproc_per_node 4 train.py --config config.py