WandbMetricsLogger

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

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

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

このコールバックは、以下のものを自動的に W&B の run ページにログ記録します。

  • システム (CPU/GPU/TPU) メトリクス
  • model.compile で定義されたトレーニングおよび検証メトリクス
  • 学習率(固定値と学習率スケジューラの両方)

注:

initial_epochmodel.fit に渡してトレーニングを再開し、学習率スケジューラを使用している場合は、initial_global_stepWandbMetricsLogger に渡してください。initial_global_stepstep_size * initial_step です。step_size は、エポックごとのトレーニングステップ数です。step_size は、トレーニングデータセットのカーディナリティとバッチサイズの積として計算できます。

引数
log_freq (“epoch”、“batch”、または int) “epoch” の場合、各エポックの最後にメトリクスをログ記録します。“batch” の場合、各バッチの最後にメトリクスをログ記録します。整数である場合、その数のバッチの最後にメトリクスをログ記録します。デフォルトは “epoch” です。
initial_global_step (int) initial_epoch からトレーニングを再開し、学習率スケジューラを使用している場合は、この引数を使用して学習率を正しくログ記録します。これは step_size * initial_step として計算できます。デフォルトは 0 です。

メソッド

set_model

set_model(
    model
)

set_params

set_params(
    params
)