ColossalAI/examples/tutorial/hybrid_parallel/test_ci.sh
Frank Lee 39163417a1
[example] updated the hybrid parallel tutorial (#2444)
* [example] updated the hybrid parallel tutorial

* polish code
2023-01-11 15:17:17 +08:00

6 lines
126 B
Bash

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