nemo_automodel.components.loggers.comet_utils

Module Contents

Classes

Name	Description
`CometLogger`	Comet ML logger for experiment tracking.

Functions

Name	Description
`build_comet`	Build a Comet logger from a raw config object.

Data

logger

API

class nemo_automodel.components.loggers.comet_utils.CometLogger(
    project_name: str,
    workspace: typing.Optional[str] = None,
    api_key: typing.Optional[str] = None,
    experiment_name: typing.Optional[str] = None,
    tags: typing.Optional[list] = None,
    auto_metric_logging: bool = False,
    kwargs = {}
)

Comet ML logger for experiment tracking.

nemo_automodel.components.loggers.comet_utils.CometLogger.__enter__()

nemo_automodel.components.loggers.comet_utils.CometLogger.__exit__(
    exc_type,
    exc_val,
    exc_tb
)

nemo_automodel.components.loggers.comet_utils.CometLogger.end() -> None

End the Comet experiment.

nemo_automodel.components.loggers.comet_utils.CometLogger.log_metrics(
    metrics: typing.Dict[str, float],
    step: typing.Optional[int] = None
) -> None

Log metrics to Comet.

Parameters:

metrics

Dict[str, float]

Dictionary of metrics to log

step

Optional[int]Defaults to None

Step number for the metrics (optional)

nemo_automodel.components.loggers.comet_utils.CometLogger.log_params(
    params: typing.Dict[str, typing.Any]
) -> None

Log parameters to Comet.

Parameters:

params

Dict[str, Any]

Dictionary of parameters to log

nemo_automodel.components.loggers.comet_utils.build_comet(
    cfg
) -> nemo_automodel.components.loggers.comet_utils.CometLogger

Build a Comet logger from a raw config object.

Back-compat shim. The logger-construction logic lives in :meth:nemo_automodel.components.loggers.loggers.CometConfig.build (the single implementation); recipes construct CometConfig via RecipeConfig.comet and call build directly. This wrapper maps a raw cfg’s comet: block onto the typed config so existing callers keep working.

Parameters:

cfg

Configuration object containing Comet settings.

Returns: CometLogger

CometLogger instance.

nemo_automodel.components.loggers.comet_utils.logger = logging.getLogger(__name__)