Fix typo in `log_graph` docs (#19254)

This commit is contained in:
awaelchli 2024-01-10 19:00:27 +01:00 committed by GitHub
parent 7a4b0fc433
commit 1bd27447d9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 2 deletions

View File

@ -15,7 +15,7 @@
- `PR9699`_
* - used Trainers flag ``checkpoint_callback``
- set ``enable_checkpointing``. If you set ``enable_checkpointing=True``, it configures a default ``ModelCheckpoint`` callback if none is provided ``lightning_pytorch.trainer.trainer.Trainer.callbacks.ModelCheckpoint``
- set ``enable_checkpointing``. If you set ``enable_checkpointing=True``, it configures a default ``ModelCheckpoint`` callback if none is provided ``lightning.pytorch.trainer.trainer.Trainer.callbacks.ModelCheckpoint``
- `PR9754`_
* - used Trainers flag ``stochastic_weight_avg``

View File

@ -61,7 +61,7 @@ Multiple loggers support visualizing the model topology. Here's an example that
.. code-block:: python
def any_lightning_module_function_or_hook(self):
tensorboard_logger = self.logger.experiment
tensorboard_logger = self.logger
prototype_array = torch.Tensor(32, 1, 28, 27)
tensorboard_logger.log_graph(model=self, input_array=prototype_array)