Distilbert Base Uncased Finetuned Imdb

🚀 harshhmaniya/distilbert-base-uncased-finetuned-imdb

該模型是基於distilbert-base-uncased在未知數據集上微調得到的版本。它在評估集上取得了如下結果：

• 訓練損失：2.7634
• 驗證損失：2.4703
• 輪數：0

🚀 快速開始

本模型是在未知數據集上對distilbert-base-uncased進行微調後的版本。如果你想使用該模型，可參考Hugging Face的transformers庫相關文檔進行加載和調用。

📄 許可證

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

🔧 技術細節

訓練超參數

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

• 優化器：{'name': 'AdamWeightDecay', 'learning_rate': {'module': 'transformers.optimization_tf', 'class_name': 'WarmUp', 'config': {'initial_learning_rate': 2e-05, 'decay_schedule_fn': {'module': 'keras.optimizers.schedules', 'class_name': 'PolynomialDecay', 'config': {'initial_learning_rate': 2e-05, 'decay_steps': -375, 'end_learning_rate': 0.0, 'power': 1.0, 'cycle': False, 'name': None}, 'registered_name': None}, 'warmup_steps': 1000, 'power': 1.0, 'name': None}, 'registered_name': 'WarmUp'}, 'decay': 0.0, 'beta_1': 0.9, 'beta_2': 0.999, 'epsilon': 1e-08, 'amsgrad': False, 'weight_decay_rate': 0.01}
• 訓練精度：混合浮點16位

訓練結果

框架版本

• Transformers 4.48.3
• TensorFlow 2.18.0
• Datasets 3.3.1
• Tokenizers 0.21.0