Downloads · 30 days
18
8% of all-time downloads
IsmaelMousa/arab-bart-base-174M
arab-bart-base-174M is a summarization model from IsmaelMousa. Use it when you need a shorter version of a longer text. It is set up for PyTorch. The card lists the license as mit.
Implemented the BART: Denoising Sequence-to-Sequence Pre-training for Natural Language Generation, Translation, and Comprehension paper from scratch using PyTorch for an abstractive summarization task in Arabic.
Downloads · 30 days
18
8% of all-time downloads
All-time downloads
222
Public
Repo size
701 MB
Likes
0
Public
Click a slice to open those files.
.pt701 MB · 99%
From the Hugging Face model README
Implemented the BART: Denoising Sequence-to-Sequence Pre-training for Natural Language Generation, Translation, and Comprehension
paper from scratch using PyTorch for an abstractive summarization task in Arabic.
[!IMPORTANT] The model inferenc is not ready, i mean you can't loading it directly from the
Transformerslibrary.As soon as possible i will create an inference API, and integrate the model with the Transformers library.
Reproduce the BART model from scratch to understand its architecture in depth, using the minimum available resources.
The model size: 174M parameters.
Abstractive Summarization in Arabic.
The dataset used is the XL-Sum(Arabic Subset) dataset. I chose this dataset because it's well-suited for our task. Additionally, it's written in pure Arabic, which makes it the best choice. The original source: BBC Arabic.
Features (columns):
Size:
32,473 rows.4689 rows.4689 rows.| Epoch | Loss(train) | Loss(validation) | Epoch Time (hours) | Training Time (hours) | Device |
|---|---|---|---|---|---|
| 1 | 10.03 | 9.72 | 0.23 | 1.1 | 1 x L4OS |
| 2 | 9.61 | 9.44 | 0.22 | 1.1 | 1 x L4OS |
| 3 | 9.36 | 9.22 | 0.22 | 1.1 | 1 x L4OS |
| 4 | 9.16 | 9.05 | 0.22 | 1.1 | 1 x L4OS |
| 5 | 9.01 | 8.92 | 0.22 | 1.1 | 1 x L4OS |
This model is licensed under the MIT License.