mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-07-17 00:41:39 +00:00
parent
55424a16a5
commit
a4fc125c34
@ -1,7 +1,7 @@
|
|||||||
set -x
|
set -x
|
||||||
export BS=${BS:-16}
|
export BS=${BS:-16}
|
||||||
export MEMCAP=${MEMCAP:-0}
|
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 MODEL=${MODEL:-"125m"}
|
||||||
export GPUNUM=${GPUNUM:-1}
|
export GPUNUM=${GPUNUM:-1}
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user