Quick Started

The SageMaker Experiments logger can be easily applied to the PyTorch Lightning Trainer class for model training:

from experiments_addon.logger import SagemakerExperimentsLogger

logger = SagemakerExperimentsLogger(experiment_name="TestExp", run_name="TestRun")
trainer = Trainer(
    logger=logger,
    ...
)
trainer.fit(...)

A more detailed demonstration of the usage of the SagemakerExperimentsLogger class can be found in the notebook: example/tutorial.ipynb.