robinsmits's picture
Update README.md
83ca45d verified
|
raw
history blame
2.96 kB
metadata
license: cc-by-nc-4.0
library_name: peft
tags:
  - trl
  - sft
  - conversational
  - generated_from_trainer
  - qwen2
base_model: Qwen/Qwen1.5-7B-Chat
model-index:
  - name: Qwen1.5-7B-Dutch-Chat-Sft
    results: []
language:
  - nl
datasets:
  - BramVanroy/ultrachat_200k_dutch
pipeline_tag: text-generation
inference: false

Qwen1.5-7B-Dutch-Chat-Sft

This model is a fine-tuned version of Qwen/Qwen1.5-7B-Chat on an unknown dataset. It achieves the following results on the evaluation set:

  • Loss: 1.1756

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: 0.0003
  • train_batch_size: 2
  • eval_batch_size: 4
  • seed: 42
  • gradient_accumulation_steps: 32
  • total_train_batch_size: 64
  • optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
  • lr_scheduler_type: cosine
  • lr_scheduler_warmup_ratio: 0.05
  • training_steps: 1466

Training results

Training Loss Epoch Step Validation Loss
1.4493 0.2 146 1.4083
1.3638 0.4 292 1.3213
1.3146 0.6 438 1.2731
1.2756 0.8 584 1.2400
1.2533 1.0 730 1.2165
1.2388 1.19 876 1.1998
1.2246 1.39 1022 1.1882
1.211 1.59 1168 1.1802
1.204 1.79 1314 1.1763
1.2041 1.99 1460 1.1756

Framework versions

  • PEFT 0.9.0
  • Transformers 4.38.2
  • Pytorch 2.2.1+cu121
  • Datasets 2.17.1
  • Tokenizers 0.15.2

Citation

Thanks to the creators of Qwen1.5 for there great work!

@article{qwen,
  title={Qwen Technical Report},
  author={Jinze Bai and Shuai Bai and Yunfei Chu and Zeyu Cui and Kai Dang and Xiaodong Deng and Yang Fan and Wenbin Ge and Yu Han and Fei Huang and Binyuan Hui and Luo Ji and Mei Li and Junyang Lin and Runji Lin and Dayiheng Liu and Gao Liu and Chengqiang Lu and Keming Lu and Jianxin Ma and Rui Men and Xingzhang Ren and Xuancheng Ren and Chuanqi Tan and Sinan Tan and Jianhong Tu and Peng Wang and Shijie Wang and Wei Wang and Shengguang Wu and Benfeng Xu and Jin Xu and An Yang and Hao Yang and Jian Yang and Shusheng Yang and Yang Yao and Bowen Yu and Hongyi Yuan and Zheng Yuan and Jianwei Zhang and Xingxuan Zhang and Yichang Zhang and Zhenru Zhang and Chang Zhou and Jingren Zhou and Xiaohuan Zhou and Tianhang Zhu},
  journal={arXiv preprint arXiv:2309.16609},
  year={2023}
}