Bitnet B1 58 3B

🚀 BitNet b1.58再現プロジェクト

このプロジェクトは、BitNet b1.58論文の再現結果です。モデルは、RedPajamaデータセットを用いて100Bトークンで学習されています。ハイパーパラメータ、二段階学習率、および重み減衰は、この論文で提案されている通りに実装されています。すべてのモデルは、このリポジトリでオープンソースとして公開されています。リソースが許す限り、より大規模なモデルやより多くのトークンでの学習を行う予定です。

📚 詳細ドキュメント

🔍 結果

PPLとゼロショット精度:

報告された数値と再現結果の差は、学習データの処理、シード、またはその他のランダムな要因によるバリエーションである可能性があります。

💻 評価

評価パイプラインは論文の著者によるものです。以下は評価を実行するためのコマンドです。

基本的な使用法

pip install lm-eval==0.3.0

高度な使用法

python eval_ppl.py --hf_path 1bitLLM/bitnet_b1_58-3B --seqlen 2048

python eval_task.py --hf_path 1bitLLM/bitnet_b1_58-3B \
    --batch_size 1 \
    --tasks \
    --output_path result.json \
    --num_fewshot 0 \
    --ctx_size 2048

📄 ライセンス

このプロジェクトはMITライセンスの下で公開されています。