Downloads · 30 days
17
34% of all-time downloads
megatron-gpt2-classification
MohammadOthman/megatron-gpt2-classification
megatron-gpt2-classification is a text classification model from MohammadOthman. Use it when you need a label for a piece of text. It is set up for transformers. The card lists the license as mit.
The megatron-gpt2-classification model is a language model trained using Megatron and Accelerate frameworks. It has been fine-tuned for classification tasks and benefits from distributed training across 4 GPUs (RTX 40…
All-time downloads
50
Public
Parameters
124M
1.5 GB on disk
Likes
0
Public
Repo makeup
Click a slice to open those files.
.pt996 MB · 67%
At a glance
- Task
- Text Classification
- Library
- transformers
- License
- mit
- Model type
- gpt2
- Access
- Public
- Created
- May 4, 2024
- Updated
- May 4, 2024
- SHA
- 42e48473
Try a prompt
Model card
From the Hugging Face model README
Megatron-GPT2-Classification
Description
The megatron-gpt2-classification model is a language model trained using Megatron and Accelerate frameworks. It has been fine-tuned for classification tasks and benefits from distributed training across 4 GPUs (RTX 4070).
Key Features
- Trained with Megatron and Accelerate.
- Distributed training on 4 GPUs (RTX 4070).
- Fine-tuned for classification tasks.
- Task
- Text Classification
- Library
- transformers
- Type
- gpt2
- License
- mit
- Languages
- en
- Created
- May 4, 2024
- Updated
- May 4, 2024