mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-05-02 05:35:29 +00:00
9 lines
274 B
Bash
9 lines
274 B
Bash
#!/bin/bash
|
|
set -euxo pipefail
|
|
|
|
pip install -r requirements.txt
|
|
|
|
# run test
|
|
colossalai run --nproc_per_node 4 --master_port 29500 train.py --config config.py --optimizer lars
|
|
colossalai run --nproc_per_node 4 --master_port 29501 train.py --config config.py --optimizer lamb
|