メインコンテンツまでスキップ

watch

GitHubでソースを見る

トーチモデルにフックして、勾配とトポロジーを収集します。

watch(
models,
criterion=None,
log: Optional[Literal['gradients', 'parameters', 'all']] = "gradients",
log_freq: int = 1000,
idx: Optional[int] = None,
log_graph: bool = (False)
)

任意のMLモデルを受け入れるように拡張すべきです。

引数説明
models(torch.Module) フックするモデル。タプルであってもよい
criterion(torch.F) 最適化されるオプションの損失値
log(str) "gradients"、"parameters"、"all"、または None のいずれか
log_freq(int) Nバッチごとに勾配とパラメータをログする
idx(int) 複数のモデルでwandb.watchを呼び出すときに使用されるインデックス
log_graph(boolean) グラフトポロジーをログする
返り値

| :--- | :--- |

| wandb.Graph | 最初のbackwardパスの後にデータが入るグラフオブジェクト |

| 例外 | |

| :--- | :--- |

| ValueError | wandb.initが呼ばれる前に呼び出された場合、またはどのモデルもtorch.nn.Moduleでない場合。 |

Was this page helpful?👍👎