watch
トーチモデルにフックして、勾配とトポロジーを収集します。
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でない場合。 |