diff --git a/pytorch_lightning/logging/comet.py b/pytorch_lightning/logging/comet.py index 81cb8940..fbf4f839 100644 --- a/pytorch_lightning/logging/comet.py +++ b/pytorch_lightning/logging/comet.py @@ -1,50 +1,3 @@ -""" -.. code-block:: python - - # ------------------- - # ONLINE MODE - # ------------------- - from pytorch_lightning.logging import CometLogger - - # arguments made to CometLogger are passed on to the comet_ml.Experiment class - comet_logger = CometLogger( - api_key=os.environ["COMET_KEY"], - workspace=os.environ["COMET_WORKSPACE"], # Optional - project_name="default_project", # Optional - rest_api_key=os.environ["COMET_REST_KEY"], # Optional - experiment_name="default" # Optional - ) - trainer = Trainer(logger=comet_logger) - - # ------------------- - # OFFLINE MODE - # ------------------- - from pytorch_lightning.logging import CometLogger - # arguments made to CometLogger are passed on to the comet_ml.Experiment class - comet_logger = CometLogger( - save_dir=".", - workspace=os.environ["COMET_WORKSPACE"], # Optional - project_name="default_project", # Optional - rest_api_key=os.environ["COMET_REST_KEY"], # Optional - experiment_name="default" # Optional - ) - trainer = Trainer(logger=comet_logger) - - -Use the logger anywhere in you LightningModule as follows: - -.. code-block:: python - - def train_step(...): - # example - self.logger.experiment.whatever_comet_ml_supports(...) - - def any_lightning_module_function_or_hook(...): - self.logger.experiment.whatever_comet_ml_supports(...) - - -""" - from logging import getLogger try: