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.