メインコンテンツへスキップ

View the source code

 システムメトリクスを 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 はトレーニングデータセットの基数とバッチサイズの積として計算できます。
引数Description
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
)
⌘I