metadata
license: other
base_model: facebook/opt-350m
tags:
- trl
- sft
- generated_from_trainer
model-index:
- name: results
results: []
results
This model is a fine-tuned version of facebook/opt-350m on CodeAlpaca-20k with 601 rows.
Model description
More information needed
Intended uses & limitations
- this use only 601 data from CodeAlpaca-20k so need to train further.
- use tag "### Question: " for asking/instruct and it will generate "### Answer: " for inference.
- the dataset is about coding with various languages, so need to train with specify programming language.
Training and evaluation data
More information needed
Training procedure
Training hyperparameters
The following hyperparameters were used during training:
- learning_rate: 1e-05
- train_batch_size: 2
- eval_batch_size: 8
- seed: 42
- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
- lr_scheduler_type: cosine
- lr_scheduler_warmup_ratio: 0.03
- num_epochs: 3
Framework versions
- Transformers 4.38.2
- Pytorch 2.2.1+cu121
- Datasets 2.18.0
- Tokenizers 0.15.2