ColossalAI/examples/tutorial/large_batch_optimizer/test_ci.sh

#!/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
[example] updated large-batch optimizer tutorial (#2448) * [example] updated large-batch optimizer tutorial * polish code * polish code 2023-01-11 08:27:31 +00:00			`#!/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`