WandbMetricsLogger

システムメトリクスを W&B に送信するロガー。

WandbMetricsLogger(
    log_freq: Union[LogStrategy, int] = "epoch",
    initial_global_step: int = 0,
    *args,
    **kwargs
) -> None

WandbMetricsLogger は、コールバックメソッドが引数として取る logs 辞書を自動的に wandb にログします。

このコールバックは、次の情報を W&B の run ページに自動的にログします:

• システム (CPU/GPU/TPU) メトリクス、
• model.compile で定義されたトレーニングと検証メトリクス、
• 学習率（固定値または学習率スケジューラのどちらも）

注釈:

initial_epoch を model.fit に渡してトレーニングを再開し、かつ学習率スケジューラを使用している場合、initial_global_step を WandbMetricsLogger に渡すことを確認してください。initial_global_stepは step_size * initial_step であり、ここで step_size はエポックごとのトレーニングステップ数です。step_size はトレーニングデータセットの基数とバッチサイズの積として計算できます。

メソッド

set_model

set_model(
    model
)

set_params

set_params(
    params
)