NeptuneLogger

NeptuneLogger#

class braintools.trainer.NeptuneLogger(project, api_token=None, name=None, tags=None, description=None, **kwargs)#

Neptune.ai logging backend.

Parameters:

  • project (str) – Neptune project name (workspace/project).

  • api_token (str | None) – Neptune API token. If None, uses NEPTUNE_API_TOKEN env variable.

  • name (str | None) – Run name.

  • tags (List[str] | None) – Tags for the run.

  • description (str | None) – Run description.

Examples

>>> logger = NeptuneLogger(project='workspace/project')
>>> trainer = Trainer(logger=logger)
finalize(status='success')[source]#

Finalize logging (called at end of training).

Parameters:

status (str) – Final status (‘success’, ‘failed’, ‘interrupted’).

log_hyperparams(params)[source]#

Log hyperparameters.

Parameters:

params (Dict[str, Any]) – Dictionary of hyperparameter names to values.

log_metrics(metrics, step=None)[source]#

Log metrics.

Parameters:

  • metrics (Dict[str, float]) – Dictionary of metric names to values.

  • step (int | None) – Global step number.

save()[source]#

Save/flush any buffered data.