mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-07-16 08:27:44 +00:00
parent
55424a16a5
commit
a4fc125c34
@ -1,7 +1,7 @@
|
||||
set -x
|
||||
export BS=${BS:-16}
|
||||
export MEMCAP=${MEMCAP:-0}
|
||||
# Acceptable values include `125m`, `350m`, `1.3b`, `2.7b`, `6.7`, `13b`, `30b`, `66b`. For `175b`
|
||||
# Acceptable values include `125m`, `350m`, `1.3b`, `2.7b`, `6.7b`, `13b`, `30b`, `66b`. For `175b`
|
||||
export MODEL=${MODEL:-"125m"}
|
||||
export GPUNUM=${GPUNUM:-1}
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user