wandb/examples

Example deep learning projects that use wandb's features.

wandb/examples – Weights & Biases を使用した実行追跡のための即実行コード

何であるか – このリポジトリは、Weights & Biases (W&B) ライブラリを一般的な機械学習ワークフローに統合する方法を示す短いスクリプトと Colab ノートブックのコレクションです。例は多くの人気フレームワーク(PyTorch、TensorFlow/Keras、Hugging Face Transformers、PyTorch Lightning、XGBoost、scikit‑learn など)をカバーし、W&B の主要機能を示しています:メトリクスのログ記録、ハイパーパラメータの保存、データセットの追跡、実行の可視化、モデルアーティファクトの管理。

なぜ重要か – W&B は実行追跡、モデルバージョン管理、共同レポートのためのプラットフォームです。これらの例を確認することで、最小限のコードで次のように実現できます:

  • 実行の開始(wandb.init)とプロジェクト名の付与。
  • 設定値の記録(wandb.config)。
  • スカラー値のメトリクスのログ記録(wandb.log)。
  • モデルの勾配、重み、構造を自動でキャプチャ(run.watch)。
  • フレームワーク固有のヘルパーの使用(例:wandb.keras.WandbMetricsLogger、XGBoost 用の WandbCallback、PyTorch Lightning 用の WandbLogger)。
  • W&B Artifacts を使ってデータセットやモデルチェックポイントを保存・取得。
  • 実行を明確に終了(wandb.finish)。

リポジトリの主な構成

  • examples/ – 各フレームワーク用の即実行 Python スクリプト。トレーニングループをログ記録するのに必要な最小限のコードを示します。
  • colabs/ – ローカル環境のセットアップなしで、ブラウザ上で統合を試せる Google‑Colab ノートブック。
  • README に短い 「はじめに」 セクションがあり、wandb パッケージのインストール、ログイン、10ステップの損失をログ記録する簡単な例の実行方法をステップバイステップで説明しています。

一般的なユーザー

  • 既存のプロジェクトに実行追跡を追加したいデータサイエンティストや ML エンジニア。
  • 新しいフレームワークを学んでいる学生で、実行ログの記録方法を具体的に見たい人。
  • ハイパーパラメータスイープ、モデル性能、リソース使用量を比較するための共有ダッシュボードが必要なチーム。

開始方法

  1. コアライブラリのインストール:
    pip install wandb
    wandb login   # wandb.com から取得した API キーを貼り付けます
    
  2. お使いのスタックに合った例を選択(例:examples/pytorch/train.py または colabs/keras.ipynb)。
  3. スクリプトまたはノートブックを実行。新しい実行が W&B ダッシュボードに表示され、ログ記録されたメトリクス、設定、アーティファクトを探索できます。

見られる内容 – README には W&B ダッシュボードのスクリーンショットと短い動画ツアーへのリンクが含まれており、実際に使ってみる前に UI の雰囲気を視覚的に把握できます。


上記のすべての詳細は、リポジトリの README から直接取得されています。追加の機能は推測されていません。

関連

  • プロジェクト
  • プロジェクト
  • Dispatch
  • プロジェクト
  • プロジェクト