Skip to content

Commit

Permalink
Fix lint
Browse files Browse the repository at this point in the history
Signed-off-by: Sid Murching <sid.murching@databricks.com>
  • Loading branch information
smurching committed Nov 8, 2020
1 parent 45b037f commit 5bdf0c9
Showing 1 changed file with 4 additions and 2 deletions.
6 changes: 4 additions & 2 deletions mlflow/pytorch/__init__.py
Expand Up @@ -617,11 +617,13 @@ def autolog(log_every_n_epoch=1):
Automatically log metrics, params, and models from `PyTorch Lightning
<https://pytorch-lightning.readthedocs.io/en/latest>`_ model training.
Autologging is performed when you call the `fit` method of
`pytorch_lightning.Trainer() <https://pytorch-lightning.readthedocs.io/en/latest/trainer.html#>`_.
`pytorch_lightning.Trainer() \
<https://pytorch-lightning.readthedocs.io/en/latest/trainer.html#>`_.
**Note**: Autologging is only supported for PyTorch Lightning models,
i.e. models that subclass
`pytorch_lightning.LightningModule <https://pytorch-lightning.readthedocs.io/en/latest/lightning_module.html>`_.
`pytorch_lightning.LightningModule \
<https://pytorch-lightning.readthedocs.io/en/latest/lightning_module.html>`_.
In particular, autologging support for vanilla Pytorch models that only subclass
`torch.nn.Module <https://pytorch.org/docs/stable/generated/torch.nn.Module.html>`_
is not yet available.
Expand Down

0 comments on commit 5bdf0c9

Please sign in to comment.