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
ModelZoo
ResNet50_tensorflow
Commits
525af938
Commit
525af938
authored
Oct 12, 2021
by
Hongkun Yu
Committed by
A. Unique TensorFlower
Oct 12, 2021
Browse files
Add BigBird base checkpoint. Fix documentation.
PiperOrigin-RevId: 402627099
parent
6a55ecde
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
7 additions
and
0 deletions
+7
-0
official/nlp/projects/bigbird/README.md
official/nlp/projects/bigbird/README.md
+7
-0
No files found.
official/nlp/projects/bigbird/README.md
View file @
525af938
...
@@ -76,6 +76,7 @@ python3 official/nlp/train.py \
...
@@ -76,6 +76,7 @@ python3 official/nlp/train.py \
--config_file
=
experiments/glue_mnli_matched.yaml
\
--config_file
=
experiments/glue_mnli_matched.yaml
\
--params_override
=
task.init_checkpoint
=
${
INIT_CKPT
}
\
--params_override
=
task.init_checkpoint
=
${
INIT_CKPT
}
\
--params_override
=
runtime.distribution_strategy
=
tpu
\
--params_override
=
runtime.distribution_strategy
=
tpu
\
--params_override
=
task.train_data.input_path
=
${
TRAIN_FILE
}
,task.validation_data.input_path
=
${
EVAL_FILE
}
\
--tpu
=
???
\
--tpu
=
???
\
--mode
=
train_and_eval
--mode
=
train_and_eval
```
```
...
@@ -97,3 +98,9 @@ python3 official/nlp/train.py \
...
@@ -97,3 +98,9 @@ python3 official/nlp/train.py \
--tpu
=
???
\
--tpu
=
???
\
--mode
=
train_and_eval
--mode
=
train_and_eval
```
```
## Checkpoints
Model | Configuration | Training Data | Checkpoint | Metrics
------------ | :--------------------------------------: | ------------------------------------------------------: | :----------------------------------------------------------------------------------------------------: | :-----:
BigBird base | 12 layer, 1024<= sequence length <= 4096 | Wiki + Books + CC-News + Stories (part of Common Crawl) |
[
bigbird_base
](
https://storage.googleapis.com/tf_model_garden/nlp/bigbird/bigbird.etc.base.keras.tar.gz
)
| Squad v1 F1 91.3, TriviaQA F1 79.8
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