Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
deepspeed
Commits
838f53b7
"vscode:/vscode.git/clone" did not exist on "a450cf81c11d206983f209586f38db06e8df2801"
Unverified
Commit
838f53b7
authored
Sep 01, 2020
by
Shaden Smith
Committed by
GitHub
Sep 01, 2020
Browse files
Switches BBS example to use mbsize=3 and gas=2 to fit in 16GB of memory. (#341)
parent
e8dd47df
Changes
5
Show whitespace changes
Inline
Side-by-side
Showing
5 changed files
with
5 additions
and
5 deletions
+5
-5
tests/model/BingBertSquad/deepspeed_bsz24_fp16_config.json
tests/model/BingBertSquad/deepspeed_bsz24_fp16_config.json
+1
-1
tests/model/BingBertSquad/deepspeed_bsz24_fp16_zero2_config.json
...odel/BingBertSquad/deepspeed_bsz24_fp16_zero2_config.json
+1
-1
tests/model/BingBertSquad/deepspeed_bsz24_fp32_config.json
tests/model/BingBertSquad/deepspeed_bsz24_fp32_config.json
+1
-1
tests/model/BingBertSquad/run_BingBertSquad.sh
tests/model/BingBertSquad/run_BingBertSquad.sh
+1
-1
tests/model/BingBertSquad/run_BingBertSquad_sanity.sh
tests/model/BingBertSquad/run_BingBertSquad_sanity.sh
+1
-1
No files found.
tests/model/BingBertSquad/deepspeed_bsz24_fp16_config.json
View file @
838f53b7
{
"train_batch_size"
:
24
,
"train_micro_batch_size_per_gpu"
:
6
,
"train_micro_batch_size_per_gpu"
:
3
,
"steps_per_print"
:
1
,
"optimizer"
:
{
"type"
:
"Adam"
,
...
...
tests/model/BingBertSquad/deepspeed_bsz24_fp16_zero2_config.json
View file @
838f53b7
{
"train_batch_size"
:
24
,
"train_micro_batch_size_per_gpu"
:
6
,
"train_micro_batch_size_per_gpu"
:
3
,
"steps_per_print"
:
1
,
"optimizer"
:
{
"type"
:
"Adam"
,
...
...
tests/model/BingBertSquad/deepspeed_bsz24_fp32_config.json
View file @
838f53b7
{
"train_batch_size"
:
24
,
"train_micro_batch_size_per_gpu"
:
6
,
"train_micro_batch_size_per_gpu"
:
3
,
"steps_per_print"
:
1
,
"optimizer"
:
{
"type"
:
"Adam"
,
...
...
tests/model/BingBertSquad/run_BingBertSquad.sh
View file @
838f53b7
...
...
@@ -121,7 +121,7 @@ echo "deepspeed: ${enable_deepspeed}"
echo
"other_args:
${
other_args
}
"
EFFECTIVE_BATCH_SIZE
=
${
batch_size
}
MAX_GPU_BATCH_SIZE
=
6
MAX_GPU_BATCH_SIZE
=
3
PER_GPU_BATCH_SIZE
=
$((
EFFECTIVE_BATCH_SIZE/num_gpus
))
if
[[
$PER_GPU_BATCH_SIZE
-lt
$MAX_GPU_BATCH_SIZE
]]
;
then
GRAD_ACCUM_STEPS
=
1
...
...
tests/model/BingBertSquad/run_BingBertSquad_sanity.sh
View file @
838f53b7
...
...
@@ -122,7 +122,7 @@ echo "deepspeed: ${enable_deepspeed}"
echo
"other_args:
${
other_args
}
"
EFFECTIVE_BATCH_SIZE
=
${
batch_size
}
MAX_GPU_BATCH_SIZE
=
6
MAX_GPU_BATCH_SIZE
=
3
PER_GPU_BATCH_SIZE
=
$((
EFFECTIVE_BATCH_SIZE/num_gpus
))
if
[[
$PER_GPU_BATCH_SIZE
-lt
$MAX_GPU_BATCH_SIZE
]]
;
then
GRAD_ACCUM_STEPS
=
1
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment