Codeparrot Ds Distilgpt2

🚀 codeparrot-ds-distilgpt2

該模型是 distilgpt2 在未知數據集上的微調版本。

🚀 快速開始

本模型是基於 distilgpt2 在未知數據集上微調得到的，可用於相關代碼生成等任務。

📄 許可證

本模型使用的是 Apache-2.0 許可證。

🔧 技術細節

訓練超參數

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

• 學習率（learning_rate）：0.0005
• 訓練批次大小（train_batch_size）：32
• 評估批次大小（eval_batch_size）：32
• 隨機種子（seed）：42
• 梯度累積步數（gradient_accumulation_steps）：8
• 總訓練批次大小（total_train_batch_size）：256
• 優化器（optimizer）：Adam，其中 betas=(0.9, 0.999)，epsilon=1e-08
• 學習率調度器類型（lr_scheduler_type）：餘弦（cosine）
• 學習率調度器熱身步數（lr_scheduler_warmup_steps）：1000
• 訓練輪數（num_epochs）：1
• 混合精度訓練（mixed_precision_training）：原生自動混合精度（Native AMP）

框架版本

• Transformers：4.39.3
• Pytorch：2.2.2+cu121
• Datasets：2.18.0
• Tokenizers：0.15.2

模型信息表格