ColossalAI/examples/tutorial/hybrid_parallel/test_ci.sh

6 lines
136 B
Bash

#!/bin/bash
set -euxo pipefail
pip install -r requirements.txt
torchrun --standalone --nproc_per_node 4 train.py --config config.py -s