mirror of https://github.com/hpcaitech/ColossalAI
5 lines
84 B
Bash
5 lines
84 B
Bash
|
for GPUNUM in 2 1
|
||
|
|
do
|
||
|
|
env BS=2 MODEL="125m" GPUNUM=$GPUNUM bash ./run_gemini.sh
|
||
|
|
done
|