Chatgpt Prompts Bart Long

🚀 ChatGPT提示生成器

本模型是基於BART-large在ChatGPT提示數據集上進行微調的版本。它在評估集上取得了以下結果：

• 訓練損失：2.8329
• 驗證損失：2.5015
• 輪數：4

🚀 快速開始

本模型可用於生成ChatGPT角色。只需像下面這樣輸入一個角色：

from transformers import BartForConditionalGeneration, BartTokenizer

example_english_phrase = "photographer"
batch = tokenizer(example_english_phrase, return_tensors="pt")
generated_ids = model.generate(batch["input_ids"], max_new_tokens=150)
output = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)

✨ 主要特性

• 基於BART - large模型微調，適用於ChatGPT提示生成。
• 可根據輸入生成ChatGPT角色。

💻 使用示例

基礎用法

from transformers import BartForConditionalGeneration, BartTokenizer

example_english_phrase = "photographer"
batch = tokenizer(example_english_phrase, return_tensors="pt")
generated_ids = model.generate(batch["input_ids"], max_new_tokens=150)
output = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)

🔧 技術細節

訓練超參數

訓練過程中使用了以下超參數：

• 優化器：{'name': 'AdamWeightDecay', 'learning_rate': 2e-05, 'decay': 0.0, 'beta_1': 0.9, 'beta_2': 0.999, 'epsilon': 1e-07, 'amsgrad': False, 'weight_decay_rate': 0.01}
• 訓練精度：float32

訓練結果

框架版本

• Transformers 4.26.0
• TensorFlow 2.9.2
• Datasets 2.8.0
• Tokenizers 0.13.2

📄 許可證

本項目採用Apache - 2.0許可證。

信息表格