ruDialoGPT-small_10

athugodage/ruDialoGPT-small_10

ruDialoGPT-small_10 is a text generation model from athugodage. 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

21% of all-time downloads

All-time downloads

Public

Repo size

1 GB

Likes

Public

Hugging Face

Repo makeup

Click a slice to open those files.

.bin501 MB · 100%

At a glance

Task: Text Generation
Library: transformers
License: mit
Model type: gpt2
Access: Public
Created: Sep 28, 2023
Updated: Sep 28, 2023
SHA: 65dcf2d9

Try a prompt

Base models

Model card

From the Hugging Face model README

ruDialoGPT-small_10

This model is a fine-tuned version of tinkoff-ai/ruDialoGPT-small on an unknown dataset. It achieves the following results on the evaluation set:

Loss: 1.4977

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: 5e-05
train_batch_size: 1
eval_batch_size: 8
seed: 42
gradient_accumulation_steps: 2
total_train_batch_size: 2
optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
lr_scheduler_type: linear
num_epochs: 10

Training results

Training Loss	Epoch	Step	Validation Loss
No log	1.0	55	1.3283
No log	2.0	110	1.2672
No log	3.0	165	1.3883
No log	4.0	220	1.3489
No log	5.0	275	1.4106
No log	6.0	330	1.4384
No log	7.0	385	1.4511
No log	8.0	440	1.4744
No log	9.0	495	1.4796
0.9816	10.0	550	1.4977

Framework versions

Transformers 4.33.3
Pytorch 2.0.1+cu117
Datasets 2.14.5
Tokenizers 0.13.3

Task: Text Generation
Library: transformers
Type: gpt2
License: mit
Created: Sep 28, 2023
Updated: Sep 28, 2023