WandbMetricsLogger

W&B에 시스템 메트릭을 전송하는 로거입니다.

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

WandbMetricsLogger는 콜백 메소드가 인수로 받는 로그 사전을 wandb에 자동으로 기록합니다.

이 콜백은 W&B run 페이지에 다음을 자동으로 기록합니다:

• 시스템(CPU/GPU/TPU) 메트릭,
• model.compile에서 정의된 트레이닝 및 검증 메트릭,
• 학습률(고정된 값이나 학습률 스케줄러 모두에 대해서)

참고 사항:

model.fit에 initial_epoch을 전달하여 트레이닝을 재개하고 학습률 스케줄러를 사용하는 경우, WandbMetricsLogger에 initial_global_step을 전달해야 합니다. initial_global_step은 step_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
)