mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-09-06 03:20:52 +00:00
[CI] add test_ci.sh for palm, opt and gpt (#2475)
This commit is contained in:
4
examples/language/opt/test_ci.sh
Normal file
4
examples/language/opt/test_ci.sh
Normal file
@@ -0,0 +1,4 @@
|
||||
for GPUNUM in 2 1
|
||||
do
|
||||
env BS=2 MODEL="125m" GPUNUM=$GPUNUM bash ./run_gemini.sh
|
||||
done
|
Reference in New Issue
Block a user