gpt-neo-125m-finetuned-couples2

ColleenMacklin/gpt-neo-125m-finetuned-couples2

gpt-neo-125m-finetuned-couples2 is a text generation model from ColleenMacklin. Use it when you need the model to write or continue text. It is set up for transformers. The card lists the license as mit.

should probably proofread and complete it, then remove this comment. --

Downloads · 30 days

12% of all-time downloads

All-time downloads

Public

Parameters

125M

1 GB on disk

Likes

Public

Hugging Face

Repo makeup

Click a slice to open those files.

.safetensors501 MB · 99%

At a glance

Task: Text Generation
Library: transformers
License: mit
Model type: gpt_neo
Access: Public
Created: Mar 10, 2024
Updated: Mar 10, 2024
SHA: fc51fa75

Try a prompt

Base models

EleutherAI/gpt-neo-125m

Model card

From the Hugging Face model README

gpt-neo-125m-finetuned-couples2

This model is a fine-tuned version of EleutherAI/gpt-neo-125m on an unknown dataset. It achieves the following results on the evaluation set:

Loss: 2.6169

Model description

More information needed

Intended uses & limitations

More information needed

Training and evaluation data

More information needed

Training procedure

Training hyperparameters

The following hyperparameters were used during training:

learning_rate: 3e-07
train_batch_size: 8
eval_batch_size: 8
seed: 42
optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
lr_scheduler_type: linear
num_epochs: 6.0

Training results

Training Loss	Epoch	Step	Validation Loss
No log	1.0	91	2.6242
No log	2.0	182	2.6215
No log	3.0	273	2.6194
No log	4.0	364	2.6180
No log	5.0	455	2.6171
2.6194	6.0	546	2.6169

Framework versions

Transformers 4.38.2
Pytorch 2.1.0+cu121
Datasets 2.18.0
Tokenizers 0.15.2

Task: Text Generation
Library: transformers
Type: gpt_neo
License: mit
Created: Mar 10, 2024
Updated: Mar 10, 2024