core.common.logger

Copyright (c) Meta, Inc. and its affiliates.

This source code is licensed under the MIT license found in the LICENSE file in the root directory of this source tree.

Classes

Logger	Generic class to interface with various logging modules, e.g. wandb,
WandBLogger	Generic class to interface with various logging modules, e.g. wandb,
TensorboardLogger	Generic class to interface with various logging modules, e.g. wandb,

Module Contents

class core.common.logger.Logger(config)

Bases: abc.ABC

Generic class to interface with various logging modules, e.g. wandb, tensorboard, etc.

config

abstract watch(model, log_freq: int = 1000)

Monitor parameters and gradients.

log(update_dict, step: int, split: str = '')

Log some values.

abstract log_plots(plots) → None

abstract mark_preempting() → None

abstract log_summary(summary_dict: dict[str, Any]) → None

abstract log_artifact(name: str, type: str, file_location: str) → None

class core.common.logger.WandBLogger(config)

Bases: Logger

Generic class to interface with various logging modules, e.g. wandb, tensorboard, etc.

project

entity

group

watch(model, log_freq: int = 1000) → None

Monitor parameters and gradients.

log(update_dict, step: int, split: str = '') → None

Log some values.

log_plots(plots, caption: str = '') → None

log_summary(summary_dict: dict[str, Any])

mark_preempting() → None

log_artifact(name: str, type: str, file_location: str) → None

class core.common.logger.TensorboardLogger(config)

Bases: Logger

Generic class to interface with various logging modules, e.g. wandb, tensorboard, etc.

writer

watch(model, log_freq: int = 1000) → bool

Monitor parameters and gradients.

log(update_dict, step: int, split: str = '')

Log some values.

mark_preempting() → None

log_plots(plots) → None

log_summary(summary_dict: dict[str, Any]) → None

log_artifact(name: str, type: str, file_location: str) → None