aubmindlab
/

aragpt2-medium

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Metrics Training metrics Community

AUB MIND LAB commited on Jan 12, 2021

Commit

1b3a0ef

•

1 Parent(s): 612594b

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -109,7 +109,7 @@ All models are available in the `HuggingFace` model page under the [aubmindlab](
 Model | Hardware | num of examples (seq len = 1024) | Batch Size | Num of Steps | Time (in days)
  ---|:---:|:---:|:---:|:---:|:---:
 AraGPT2-base | TPUv3-128 | 9.7M | 1792 | 125K | 1.5
-AraGPT2-medium | TPUv3-8 | 9.7M | 1152 | 1M | 15
 AraGPT2-large | TPUv3-128 | 9.7M | 256 | 220k | 3
 AraGPT2-mega | TPUv3-128 | 9.7M | 256 | 780K | 9

 Model | Hardware | num of examples (seq len = 1024) | Batch Size | Num of Steps | Time (in days)
  ---|:---:|:---:|:---:|:---:|:---:
 AraGPT2-base | TPUv3-128 | 9.7M | 1792 | 125K | 1.5
+AraGPT2-medium | TPUv3-8 | 9.7M | 80 | 1M | 15
 AraGPT2-large | TPUv3-128 | 9.7M | 256 | 220k | 3
 AraGPT2-mega | TPUv3-128 | 9.7M | 256 | 780K | 9